前后缀分解

238.除自身以外数组的乘积

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

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

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

示例 1:

输入: nums = [1,2,3,4]

输出: [24,12,8,6]

思路

题目要求answer[i] 等于 nums 中除 nums[i] 之外其余各元素的乘积。我们可以将其分为左边pre和右边suf两个部分。

$$pre[i]\mathrm{表}\mathrm{示}\mathrm{从}nums[0]\mathrm{到}nums[i-1]\mathrm{的}\mathrm{乘}\mathrm{积}$$$$pre[i]\mathrm{的}\mathrm{值}\mathrm{可}\mathrm{以}\mathrm{通}\mathrm{过}\mathrm{累}\mathrm{乘}，\mathrm{使}\mathrm{用}pre[i-1]\ast sum[i-1]\mathrm{快}\mathrm{速}\mathrm{更}\mathrm{新}\mathrm{值}$$$$suf[i]\mathrm{表}\mathrm{示}\mathrm{从}nums[i+1]\mathrm{到}nums[n-1]\mathrm{的}\mathrm{乘}\mathrm{积}，suf\mathrm{可}\mathrm{以}\mathrm{使}\mathrm{用}\mathrm{相}\mathrm{同}\mathrm{方}\mathrm{法}\mathrm{更}\mathrm{新}$$$$answer[i]=pre[i]\ast suf[i]$$

class Solution {
    public int[] productExceptSelf(int[] nums) {
        int n = nums.length;
        int[] pre = new int[n];
        pre[0] = 1;
        for (int i = 1; i < n; i++) {
            pre[i] = pre[i - 1] * nums[i - 1];
        }

        int[] suf = new int[n];
        suf[n - 1] = 1;
        for (int i = n - 2; i >= 0; i--) {
            suf[i] = suf[i + 1] * nums[i + 1];
        }
        
        int[] ans = new int[n];
        for (int i = 0; i < n; i++) {
            ans[i] = pre[i] * suf[i];
        }
        return ans;
    }
}