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

xiao_fwuu2024-05-09 23:52

给你一个整数数组 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]

提示：

2 <= nums.length <= 105
-30 <= nums[i] <= 30
保证数组 nums之中任意元素的全部前缀元素和后缀的乘积都在 32 位 整数范围内

思路：

本题可采用取每个元素 nums $i$ 之前所有的元素的乘积 pre $i$ 和每个元素之后所有元素的乘积 behind $i$ 做乘积的方法，得出 res $i$ 。即 res $i$ = pre $i$ * behind $i$ ，具体可见网上大佬的解析：

代码：

java 复制代码

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

参考：. - 力扣（LeetCode）