题目:给定一个数组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[i]的计算公式,可以将B[i]的计算分成两部分,C[i]=A[0]*A[1]*...*A[i-1],D[i]=A[n-1]*...*A[i+1],B[i]=C[i]*D[n-i-1],n是数组A的长度。
class Solution {
public:
vector<int> multiply(const vector<int>& A) {
int len = A.size();
vector<int> B(len,1);
vector<int> C(len,1);
vector<int> D(len,1);
if(len == 0) return B;
for(int i=1;i<len;i++){
C[i] = A[i-1]*C[i-1];
D[i] = A[len-i]*D[i-1];
}
for(int i=0;i<len;i++){
B[i] = C[i]*D[len-i-1];
}
return B;
}
};
②讨论区看到的:按下图把每行被1分割的两部分乘积都计算出来,这样可以从首尾分别用累乘算出两个列表,然后两个列表首尾相乘就是B的元素,先算下三角中的连乘,即我们先算出B[i]中的一部分,然后倒过来按上三角中的分布规律,把另一部分也乘进去。
代码:
class Solution {
public:
vector<int> multiply(const vector<int>& A) {
int len = A.size();
vector<int> B(len,1);
if(len == 0) return B;
for(int i=1;i<len;i++){
B[i] = B[i-1]*A[i-1];
}
int temp = 1;
for(int j = len-2;j>=0;j--){
temp = temp*A[j+1];
B[j] = B[j]*temp;
}
return B;
}
};