码迷,mamicode.com
首页 > 编程语言 > 详细

除自身以外数组的乘积

时间:2021-03-04 13:33:58      阅读:0      评论:0      收藏:0      [点我收藏+]

标签:题目   ems   空间   计算   self   code   lan   变量   length   

除自身以外数组的乘积

题目:
给你一个长度为 n 的整数数组 nums,其中 n > 1,返回输出数组 output ,其中 output[i] 等于 nums 中除 nums[i] 之外其余各元素的乘积。

示例:

输入: [1,2,3,4]
输出: [24,12,8,6]

提示:题目数据保证数组之中任意元素的全部前缀元素和后缀(甚至是整个数组)的乘积都在 32 位整数范围内。

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

解题思路:可以用两个数组来分别记录从左到当前下标的乘积和从右到当前下标的乘积,之后对数组每个元素进行计算即可

class Solution {
    public int[] productExceptSelf(int[] nums) {
        int len = nums.length;
        int[] dp_l = new int[len];
        int[] dp_r = new int[len];
        int[] output = new int[len];
        int sum = 1;
        for(int i = 0; i < len; i++) {
            sum *= nums[i];
            dp_l[i] = sum;
        }
        
        sum = 1;
        for(int i = len - 1; i >= 0; i--) {
            sum *= nums[i];
            dp_r[i] = sum;
        }
        
        for(int i = 0; i < len; i++) {
            if(i == 0) {
                output[i] = dp_r[1];
                continue;
            }
            if(i == len - 1) {
                output[i] = dp_l[len - 2];
                continue;
            }
            output[i] = dp_l[i - 1] * dp_r[i + 1];            
        }
        
        return output;
    }
}

解题思路2:可以用O(1)空间复杂度来解决,声明一个变量来存储[0..i - 1]的乘积然后与output数组相乘会得到当前下标左边的乘积,之后用相同方法声明变量存储[i - 1.. 0]的乘积与output数组相乘即可

class Solution {
    public int[] productExceptSelf(int[] nums) {
        int len = nums.length;
        int[] output = new int[len];
        int product = 1;
        for(int i = 0; i < len; i++) {
            output[i] = 1;
        }
        
        for(int i = 0; i < len; i++) {
            output[i] = product * output[i];
            product *= nums[i];
        }
        
        product = 1;
        for(int i = len - 1; i >= 0; i--) {
            output[i] = product * output[i];
            product *= nums[i];
        }
        
        return output;
    }
}

除自身以外数组的乘积

标签:题目   ems   空间   计算   self   code   lan   变量   length   

原文地址:https://www.cnblogs.com/katoMegumi/p/14479207.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!