题目描述:
给一个数组,求数组的各元素乘积(不包括当前位置)。
思路:
用两个数组记录顺序数组乘积和逆序数组乘积。
代码:
class Solution {
public:
vector<int> productExceptSelf(vector<int>& nums) {
int len = nums.size();
vector<int> pre(len);
vector<int> back(len);
pre[0] = nums[0];
for (int i=1; i<len; ++i) {
pre[i] = pre[i-1] * nums[i];
}
back[len-1] = nums[len-1];
for (int i=len-2; i>=0; --i) {
back[i] = back[i+1] * nums[i];
}
vector<int> ans(len);
for (int i=0; i<len; ++i) {
if (i == 0) ans[i] = back[i+1];
else if (i == len-1) ans[i] = pre[i-1];
else ans[i] = pre[i-1] * back[i+1];
}
return ans;
}
};