剑指offer66 构建乘积数组

剑指offer66 构建乘积数组

    给定一个数组A[0, 1, …, n-1]，请构建一个数组B[0, 1, …, n-1]，其中B中的元素B[i]=A[0]×A[1]×… ×A[i-1]×A[i+1]×…×A[n-1]。

    不能使用除法。

样例

输入：[1, 2, 3, 4, 5]
输出：[120, 60, 40, 30, 24]

思考题：

    能不能只使用常数空间？（除了输出的数组之外）

思路：

    可以想象B[i] 是由两部分乘积得到：A[0],…,A[i-1] 和A[i+1],…,A[n]

    这两部分乘积分开计算，第一部分:先从B第1个位置计算到第n-1个位置，然后mul_1不断累乘A[0]到A[i-1]，即B[i] = mul_1

    第二部分: 从B的第n-2个位置，开始计算，同样用mul_2不断累乘A[n-1]到A[0]，然后B[i] *= mul_2，累乘即为了乘上第一部分得到的。

AcWing-86 C++ code：

class Solution {
public:
    vector<int> multiply(const vector<int>& A) {
        int n = A.size();
        int mul_1 = 1;
        int mul_2 = 1;
        vector<int> B(n, 1);
        for(int i = 1; i <= n - 1; i++){
            mul_1 *= A[i - 1];
            B[i] = mul_1;
        }
        for(int i = n - 2; i >= 0; i--){
            mul_2 *= A[i + 1];
            B[i] *= mul_2;
        }
        
        return B;
    }
};