(三)剑指offer 50构建数组乘积

题目描述

给定一个数组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:
    vector<int> multiply(const vector<int>& A)
    {
        int n=A.size();
        vector<int>result;
        if(n==0)return result;
        vector<int>b1(n);
        vector<int>b2(n);
        b1[0]=b2[n-1]=1;
        for(int i=1;i<n;i++)//左下三角从上至下；
            b1[i]=b1[i-1]*A[i-1];
        for(int j=n-2;j>=0;j--)//右上三角。从下至上
            b2[j]=b2[j+1]*A[j+1];
        result=b1;
        for(int i=0;i<n;i++)
            result[i]=b1[i]*b2[i];
        return result;
    }
};