题目描述
给定一个数组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]
思考题:
- 能不能只使用常数空间?(除了输出的数组之外)
问题分析
此题先创建一个ans数组,然后遍历两次给定数组nums,第一次从左往右,第二次从右往左,就得到了结果并返回。第一次遍历是将每个位置前面的乘积保存在当前位置。第二次遍历是将每个位置后面的乘积乘以当前位置之前算出的结果,就得到了该位置的最终结果。
代码实现
class Solution {
public:
vector<int> multiply(const vector<int>& A) {
vector<int> ans(A.size(), 1);
int temp = 1;
for(int i = 1; i < A.size(); i++){
temp *= A[i-1];
ans[i] = temp;
}
temp = 1;
for(int i = A.size() - 2; i >= 0; i--){
temp *= A[i+1];
ans[i] *= temp;
}
return ans;
}
};