LeetCode 238: Product of Array Except itself

weixin_34289744

于 2017-08-22 15:31:00 发布

阅读量86

点赞数

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/shuashuashua/p/7411845.html

本文介绍了一种常数空间复杂度的解决方案，用于计算数组中除当前元素外所有其他元素的乘积。提供了两种实现方法：一种通过计数零元素并单独处理，另一种采用更优雅的方式，通过两次遍历来分别计算左侧和右侧元素的累积乘积。

Constant space solution that counting zeros:

class Solution {
    public int[] productExceptSelf(int[] nums) {
        int[] result = new int[nums.length];
        int zeros = 0;
        int multi = 1;
        
        for (int num : nums) {
            if (num == 0) {
                zeros++;
            } else {
                multi *= num;
            }
            
            if (zeros > 1) {
                return result;
            }
        }
        
        for (int i = 0; i < nums.length; i++) {
            if (zeros > 0 && nums[i] != 0) {
                continue;
            }
            
            result[i] = nums[i] == 0 ? multi : multi / nums[i];
        }
        return result;
    }
}

Nice Solution:

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