lintcode50 数组剔除元素后的乘积

数组剔除元素后的乘积

给定一个整数数组A。

定义B[i] = A[0] * ... * A[i-1] * A[i+1] * ... * A[n-1]， 计算B的时候请不要使用除法。

您在真实的面试中是否遇到过这个题？

Yes

样例

给出A=[1, 2, 3]，返回 B为[6, 3, 2]

代码

class Solution {
public:
    /*
     * @param nums: Given an integers array A
     * @return: A long long array B and B[i]= A[0] * ... * A[i-1] * A[i+1] * ... * A[n-1]
     */
    vector<long long> productExcludeItself(vector<int> &nums) {
        // write your code here

        int len = nums.size();
        vector<long long> left(len, );
        vector<long long> right(len, );
        vector<long long> res(len, );

        for (int i = ; i < len; ++i) {
            left[i] = left[i - ] * nums[i - ];
        }
        for (int i = len - ; i >= ; --i) {
            right[i] = right[i + ] * nums[i + ];
        }
        for(int i = ; i < len; ++i) {
            res[i] = left[i] * right[i];
        }
        return res;
    }
};