给定一个数组 A[0,1,…,n-1],请构建一个数组 B[0,1,…,n-1],其中 B[i] 的值是数组 A 中除了下标 i 以外的元素的积, 即 B[i]=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]=C[i]×D[i],C[i]=A[0]×A[1]×…×A[i-1],D[i]=A[i+1]×…×A[n-1]
可以先求出C和D,再求答案。
代码:
class Solution {
public:
vector<int> constructArr(vector<int>& a) {
int len = a.size();
if(len == 0) return {};
vector<int> left(len);
vector<int> right(len);
left[0] = 1;
for(int i = 1;i < len;i++) {
left[i] = a[i-1]*left[i-1];
}
right[len - 1] = 1;
for(int i = len - 2;i >= 0;i--) {
right[i] = a[i+1]*right[i+1];
}
vector<int> ans(len);
for(int i = 0;i < len;i++) {
ans[i] = left[i]*right[i];
}
return ans;
}
};