牛客网剑指offer-构建乘积数组

题目描述

给定一个数组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]。不能使用除法。

class Solution {
public:
    //把B数组的值分成两部分A[0]到A[i-1],和A[I+1]到A[n-1];
    //两部分可以分别正向和反向累乘得到
    vector<int> multiply(const vector<int>& A) {
        vector<int> B;
        int d = 1;
        for (int i = 0; i < A.size(); ++i)
        {
            B.push_back(d);
            d *= A[i];
        }
        d = 1;
        for (int i = A.size()-1; i >= 0; --i)
        {
            B[i] *= d;
            d *= A[i];
        }
        return B;
    }
};