剑指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]。不能使用除法。

思路

用O(n^2)的方法很简单么还有更好的方法
比如1 2 3 4 5
res[0]=A[1]*A[2]*A[3]*A[4]=2*3*4*5
res[1]=A[0]*A[2]*A[3]*A[4]=1*3*4*5
res[2]=A[0]*A[1]*A[3]*A[4]=1*2*4*5
res[3]=A[0]*A[1]*A[2]*A[4]=1*2*3*5
res[4]=A[0]*A[1]*A[2]*A[3]=1*2*3*4
这里把计算res的数值分成两部分放c和d数组里
c从上往下计算，d从下往上计算

code

class Solution {
public:
    vector<int> multiply(const vector<int>& A) {
        int Asize=A.size();
        vector<int> res(Asize),c(Asize),d(Asize);
        for(int i=0,j=Asize-1;i<Asize&&j>=0;++i,--j){
            if(i==0){
                c[i]=1;
                d[j]=1;
                continue;
            }
            c[i]=c[i-1]*A[i-1];
            d[j]=d[j+1]*A[j+1];
        }
        for(int i=0;i<Asize;++i){
            res[i]=c[i]*d[i];
        }
        return res;
    }
};