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

119 阅读2分钟

给你一个整数数组 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 位 整数范围内

 

进阶: 你可以在 O(1) 的额外空间复杂度内完成这个题目吗?( 出于对空间复杂度分析的目的,输出数组 不被视为 额外空间。)

原文链接:leetcode.cn/problems/pr…

题解

题目规定不能用除法,那么暴力的方法就是对当前数的前后两个方向的遍历,那无疑会花费大量时间。那可以用前缀与后缀的思想来做(分别算出前缀积和后缀积,然后再分别乘起来),不过题目又要求只开一个数组,那么只能先求出一个方向的积,剩下的积就在最后一个循环中求出。比如这里就求出了前缀积,然后定义一个变量,这个变量初始值是1(注意此时循环是从后往前),没一次运算都乘以当前的数字,作为后缀积,于此,可以得出结果。

class Solution {
    public int[] productExceptSelf(int[] nums) {
        int[] a=new int[nums.length];
        //if(nums.length>1)a[0]=nums[1];
        //int left=1,right=1;
        a[0]=nums[0];
        for(int i=1;i<nums.length-1;i++){
            a[i]=nums[i]*a[i-1];
        }
        int sum=1;
        for(int i=nums.length-1;i>=1;i--){
            int k=nums[i];
            nums[i]=sum*a[i-1];sum*=k;
            
        }
        nums[0]=sum;
        return nums;
    }
}

图片.png