【LeetCode 热题 HOT 100】238. 除自身以外数组的乘积 （C++ 前缀后缀积）

238. 除自身以外数组的乘积 - 力扣（LeetCode） (leetcode-cn.com)

题意

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

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

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

思路

题目中的两个要求：不用除法+时间复杂度O(n)

考虑利用前缀和的思想，构造前缀积数组和后缀积数组。

l[i]表示i前面的数组的数的乘积，r[i]表示i后面的数的乘积。

对于数组l，初始化l[0]=1，因为下标为0的前面没有数；然后递推l[i]=l[i-1]*nums[i-1]

对于数组r，初始化r[n-1]=1，因为下标为n-1的后面没有数；然后递推r[i]=r[i+1]*nums[i+1]

最后再遍历一遍，第i个位置的答案就是l[i]*r[i]

代码

class Solution {
public:
    vector<int> productExceptSelf(vector<int>& nums) {
        int n=nums.size();
        int l[n+1],r[n+1];
        vector<int>ans;
        l[0]=1;
        for(int i=1;i<n;i++) l[i]=l[i-1]*nums[i-1];
        r[n-1]=1;
        for(int i=n-2;i>=0;i--) r[i]=r[i+1]*nums[i+1];
        for(int i=0;i<n;i++) ans.push_back(l[i]*r[i]);
        return ans;
    }
};