剑指offer——构建乘积数组（c++)

题目描述

给定一个数组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[1] * … * A[i-1]) * (A[i+1] * … * A[n-1])
因此执行两趟循环：
第一趟正向遍历数组，计算A[0] ~ A[i-1]的乘积，left[i]=A[0]*A[1]*…*A[i-1]
第二趟反向遍历数组，计算A[i+1] ~ A[n-1]的乘积， right[i] = A[i+1]*A[i+2]*…*A[n-1]
最后结果B[i]=left[i]*right[i]。

class Solution {
public:
    vector<int> multiply(const vector<int>& A) {
        if(A.empty())
            return {};
        int n = A.size();
        vector<int> left(n, 1), right(n, 1), res(n);
        for(int i = 1; i < n; ++i){
            left[i] = left[i-1] * A[i-1];
        }
        for(int j = n-2; j >= 0; --j){
            right[j] = right[j+1] * A[j+1];
        }
        for(int i = 0; i < n; ++i){
            res[i] = left[i] * right[i];
        }
        return res;
    }
};