题目描述
给定一个数组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]。不能使用除法。(注意:规定B[0] = A[1] * A[2] * … * A[n-1],B[n-1] = A[0] * A[1] * … * A[n-2];)
思路
- 如下图所示,B[0]=A[1]*…*A[i-1]A[i+1]…*A[n-1]
B[1]=A[0]A[1]…*A[i-1]A[i+1]…*A[n-1]
```
B[i]=A[0]A[1]…*A[i-1]A[i+1]…*A[n-1]
可以看到B[i]公式中没有A[i]项,由于不能使用除法,而且每一项B[i]如果进行遍历,则复杂度是N(O^2)级别,所以可以使用分两次计算,先从上到下计算红色区域,再从上到下计算紫色区域。
具体代码
class Solution {
public:
vector<int> multiply(const vector<int>& A) {
int len=A.size();
vector<int>res(len);
if(len<1)return res;
res[0]=1;
//计算红色部分
for(int i=1;i<len;i++){
res[i]=res[i-1]*A[i-1];
}
int tmp=1;
//计算紫色区域
for(int i=len-2;i>=0;i--){
tmp*=A[i+1];
res[i]=res[i]*tmp;
}
return res;
}
};