LintCode-50.数组剔除元素后的乘积

数组剔除元素后的乘积

给定一个整数数组A。

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

样例

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

标签

LintCode 版权所有 前后遍历

code

class Solution {
public:
    /**
     * @param A: 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 n = nums.size(),i;
        vector<long long> left(n, 1);
        vector<long long> right(n, 1);
        vector<long long> res(n, 1);

        for(i = 1; i < n; ++i)
            left[i] = left[i-1] * nums[i-1];
        for(i = n-2; i >= 0; --i)
            right[i] = right[i+1] * nums[i+1];

        for(i = 0; i != nums.size(); ++i)
            res[i] = left[i] * right[i];

        return res;
    }
};