Lintcode 50. 数组剔除元素后的乘积关注问题

50. 数组剔除元素后的乘积关注问题

链接：https://www.lintcode.com/problem/product-of-array-exclude-itself/description

描述

给定一个整数数组A。
定义B[i] = A[0] * ... * A[i-1] * A[i+1] * ... * A[n-1]， 计算B的时候请不要使用除法。请输出B。

样例

样例 1

输入: A = [1, 2, 3]
输出: [6, 3, 2]
解析：B[0] = A[1] * A[2] = 6; B[1] = A[0] * A[2] = 3; B[2] = A[0] * A[1] = 2

样例 2

输入: A = [2, 4, 6]
输出: [24, 12, 8]

分析：本题要求不用除法求出除A[i]外的其他元素之积，显然如果我们用除法获得所有元素的乘积就，只要除掉该元素就可以了。但是本题约定了不能除，我们换个思路。不带A[i]位置也就是让A[1]~A[i-1]与A[i+1]~A[n]相乘。我们开两个数组，Left[i]维护A[1]~A[i-1]的乘积，Right[i]维护A[i+1]~A[R]的乘积，使用递推法维护两个数组，那么对于每个A[i]，它对应的结果都是Left[i]*Right[i]。

注：需要初始化时Left[0]=Right[n-1]=1。

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]
     */
    long long Left[5010],Right[5010];
    vector<long long> productExcludeItself(vector<int> &nums) {
        // write your code here
        int i, n = nums.size();
        vector<long long> ans;
        Left[0] = Right[n - 1] = 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 < n; ++ i){
            ans.push_back(Left[i] * Right[i]);
        }
        return ans;
    }
};