题目描述
给定一个数组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;
}
};