题目描述
给定一个数组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]。不能使用除法。
样例
输入:[1, 2, 3, 4, 5]
输出:[120, 60, 40, 30, 24]
思路
由于B[i] = (A[0] * A[1] * … * A[i-1]) * (A[i+1] * … * A[n-1])
因此执行两趟循环:
第一趟正向遍历数组,计算A[0] ~ A[i-1]的乘积,left[i]=A[0]*A[1]*…*A[i-1]
第二趟反向遍历数组,计算A[i+1] ~ A[n-1]的乘积, right[i] = A[i+1]*A[i+2]*…*A[n-1]
最后结果B[i]=left[i]*right[i]。
class Solution {
public:
vector<int> multiply(const vector<int>& A) {
if(A.empty())
return {};
int n = A.size();
vector<int> left(n, 1), right(n, 1), res(n);
for(int i = 1; i < n; ++i){
left[i] = left[i-1] * A[i-1];
}
for(int j = n-2; j >= 0; --j){
right[j] = right[j+1] * A[j+1];
}
for(int i = 0; i < n; ++i){
res[i] = left[i] * right[i];
}
return res;
}
};