题目描述
给定一个数组
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]
思考题:
- 能不能只使用常数空间?(除了输出的数组之外)
算法:
class Solution {
public:
vector<int> multiply(const vector<int>& A) {
vector<int>B(A.size());
if(A.size()==0)
return B;
B[0] = A[0];
for(int i=1; i<A.size(); i++)
{
B[i] = A[i]*B[i-1];
}
int count = 1;
for(int i=B.size()-1; i>=1; i--)
{
B[i] = B[i-1]*count;
count *= A[i];
}
B[0] = count;
return B;
}
};