构建乘积数组

题目：构建乘积数组
给定一个数组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]。不能使用除法。
时间复杂度为O(n)的解法
思路：

不妨定义C[i]=A[0]A[1]…A[i-1], D[i]=A[i+1]…*A[n-2]*A[n-1].
C[i]可以用自上而下的顺序计算出来，即C[i]=C[i-1]*A[i-1].
类似的，D[i]可以用自下而上的顺序计算出来，即D[i]=D[i+1]*A[i+1]
来自《剑指offer—名企面试官精讲典型编程题》

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

更简洁的一种写法，来自牛客网LOL
//B[i]=A[0]*A[1]*...*A[i-1]*A[i+1]*...*A[n-1]
//从左到右算 B[i]=A[0]*A[1]*...*A[i-1]
//从右到左算B[i]*=A[i+1]*...*A[n-1]
//其中左边部分从上到下依次累乘，右边部分从下到上依次累乘
class Solution {
public:
    vector<int> multiply(const vector<int>& A) {

        int n=A.size();
        vector<int> b(n);
        int ret=1;
        for(int i=0;i<n;ret*=A[i++]){
            b[i]=ret;
        }
        ret=1;
        for(int i=n-1;i>=0;ret*=A[i--]){
            b[i]*=ret;
        }
        return b;
    }
};