除自身以外数组的乘积(前缀积+后缀积)

给你一个整数数组 nums，返回 数组 answer ，其中 answer[i] 等于 nums 中除 nums[i] 之外其余各元素的乘积 。

题目数据 保证 数组 nums之中任意元素的全部前缀元素和后缀的乘积都在  32 位 整数范围内。

请 不要使用除法，且在 O(n) 时间复杂度内完成此题。

 

示例 1:

输入: nums = [1,2,3,4]
输出: [24,12,8,6]

示例 2:

输入: nums = [-1,1,0,-3,3]
输出: [0,0,9,0,0]

 

class Solution {
public:
    vector<int> productExceptSelf(vector<int>& nums) {
        int n = nums.size();
        vector<int> res(n,0);
        vector<int> temp(n,0);
        for(int i=0;i<n;i++){//从左往右，计算每个元素左边的乘积
            if(i==0){
                res[i]=1;
            }else{
                res[i]=res[i-1]*nums[i-1];
            }
        }
        for(int i=n-1;i>=0;i--){//从右往左，计算每个元素右边的乘积
            if(i==n-1){
                temp[i]=1;
            }else{
                temp[i]=temp[i+1]*nums[i+1];
            }
        }
        for(int i=0;i<n;i++){//计算结果
                res[i]=res[i]*temp[i];
        }
        return res;
    }
};