力扣-238 - 技术栈

力扣-238

桔筐2024-06-12 14:57

题目

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

思路

题目要求不要用乘法，且复杂度为On。所以我们考虑用前缀乘积的方式来做，考虑到不止有前缀还有后缀我们需要一个等大的数组记录下前缀方便后面倒序累乘后缀乘积

解题方法

建立数组temp，temp[i]表示下标为i数的前缀乘积（不包括他本身因为题目要的就是除他本身的乘积），注意第一个temp[0]=1方便后续后缀乘积的计算。记录好前缀乘积后倒序累乘，用一个suf保存倒序累乘的值，suf初始值为1，遍历从n-2开始，n-1因为已经是最后一个不用算了

代码

java 复制代码

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