给定一个数组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] = 前i - 1的乘积,然后再把后边i + 1 到n的乘积乘上,用一个变量p来存储前i-1的个数的乘积,每次p *= A[i];
C++代码:
class Solution {
public:
vector<int> multiply(const vector<int>& A) {
if(A.empty()) return vector<int>();
int n = A.size();
vector<int> B(n);
int p = 1;
for(int i = 0; i < n; i++){
B[i] = p;
p *= A[i];
}
p = 1;
for(int i = n - 1; i >= 0; i--){
B[i] *= p;
p *= A[i];
}
return B;
}
};