想了半天,所幸还是想到了。
从前往后遍历容器,用一个容器用于记录当前元素前所有数的乘积,之后再从后往前遍历,记录当前元素后所有数的乘积,当前元素的result就是两个容器对应位置的乘积。
class Solution {
public:
vector<int> productExceptSelf(vector<int>& nums) {
vector<int> result(nums.size());
vector<int> product{nums[0]};
for(int i=1;i<nums.size()-1;i++){
product.push_back(product[product.size()-1]*nums[i]);
}
int post_product=1;
for(int i=nums.size()-1;i>0;i--){
result[i]=product[i-1]*post_product;
post_product*=nums[i];
}
result[0]=post_product;
return result;
}
};
优化一下空间:
class Solution {
public:
vector<int> productExceptSelf(vector<int>& nums) {
vector<int> product(nums.size());
product[0]=nums[0];
for(int i=1;i<nums.size()-1;i++){
product[i]=product[i-1]*nums[i];
}
int post_product=1;
for(int i=nums.size()-1;i>0;i--){
product[i]=product[i-1]*post_product;
post_product*=nums[i];
}
product[0]=post_product;
return product;
}
};