leetcode-238. Product of Array Except Self

238. Product of Array Except Self

Given an array of n integers where n > 1, nums, return an array output such that output[i] is equal to the product of all the elements of nums except nums[i].

Solve it without division and in O(n).

For example, given [1,2,3,4], return [24,12,8,6].

Follow up:
Could you solve it with constant space complexity? (Note: The output array does not count as extra space for the purpose of space complexity analysis.)

题意：

给定一个数组nums，结果返回另一个数组output，output[i] 是nums 数组里除了 nums[i] 元素的其他元素的积。

（是否能使时间复杂度为 O(n) ）

题解：

自己一开始的想法是先求出 nums 里所有元素的积 product，在用product/nums[i] 作为 output[i]

但是提交会发现 [0,0] ,[1,0] 等情况就会出现 除0 等异常，而且边界情况太多，单独考虑太麻烦，应该就是方法有问题。

看了 discuss 的方法，发现真是巧妙：

设 n = nums.length 

对于结果output数组，先使output[i] 等于 nums 的前 i-1 个元素的积，则output的最后一个元素的就为所求的结果。

而对于前 n -1 个元素，output[i] 应该等于  output[i] * （后面 n - i 个元素的积）

例如：

第一遍循环:   (output[0]= 1)                   

nums:    [ 1 , 2 , 3 , 4 , 5]

output:   [ 1, 1 , 2 , 6 , 24]

第二遍循环：

Output[5]= output[5]  (即24）

Output[4]= output[4] * nums[5] （即6*5）

Output[3]= output[3] * nums[5]*nums[4] （即2 *5 *4）

…  

public class Solution {
    public int[] productExceptSelf(int[] nums) {
        int n = nums.length;
        int[] res = new int[n];
        res[0] = 1;
        for (int i = 1; i < n; i++) {
            res[i] = res[i - 1] * nums[i - 1];
        }
        int right = 1;
        for (int i = n - 1; i >= 0; i--) {
            res[i] *= right;
            right *= nums[i]; //保存i ~ n 的积
        }
        return res;
    }
}