lintcode之 数组剔除元素后的乘积

题目:

给定一个整数数组A。

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

解答：

保存一个左数组和右数组即可

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 size = nums.size();
        if (size == 0) {
            return {};
        }
        vector<long long> res(size, 0);
        vector<long long> left(size, 1);
        vector<long long> right(size, 1);
        for (int i = 1; i < size; ++i) {
            left[i] = left[i - 1] * nums[i - 1];//不包含当前数字的左边
        }
        for (int j = size - 2; j >= 0; j--) {
            right[j] = right[j + 1] * nums[j + 1];
        }
        for (int i = 0 ;i < size; ++i) {
            long long int l = (i == 0) ? 1 : left[i];
            long long int r = (i == size - 1) ? 1 : right[i];
            res[i] = l * r;
        }
        return res;
    }
};