【Lintcode】1310. Product of Array Except Self

题目地址：

https://www.lintcode.com/problem/product-of-array-except-self/description

给定一个数组$A$，要求返回一个数组$B$，使得$B[i]={\prod }_{j\ne i}A[j]$。要求不能用除法，时间$O(n)$。

思路是先预处理一下前缀积和后缀积，这样$B[i]=pr[i-1]\times su[i+1]$，这里$pr[i]=A[0]...A[i]$，$su[i]=A[i]...A[l_A-1]$。代码如下：

public class Solution2 {
    /**
     * @param nums: an array of integers
     * @return: the product of all the elements of nums except nums[i].
     */
    public int[] productExceptSelf(int[] nums) {
        // write your code here
        int[] left = new int[nums.length], right = new int[nums.length];
        
        for (int i = 0; i < nums.length; i++) {
            left[i] = i == 0 ? nums[0] : nums[i] * left[i - 1];
        }
        for (int i = nums.length - 1; i >= 0; i--) {
            right[i] = i == nums.length - 1 ? nums[nums.length - 1] : nums[i] * right[i + 1];
        }
        
        int[] res = new int[nums.length];
        for (int i = 0; i < nums.length; i++) {
            int prod = 1;
            if (i >= 1) {
                prod *= left[i - 1];
            }
            if (i <= nums.length - 2) {
                prod *= right[i + 1];
            }
            res[i] = prod;
        }
        
        return res;
    }
}

时空复杂度$O(n)$。

如果可以用除法的话，可以先遍历数组$A$，存储一下$A$的所有非零数的乘积$P$、零的个数以及零的下标$r$（零的下标只需要存其中一个就行）。接着就可以构造数组$B$了。显然，如果零的个数大于$1$，那么$B$就是个零数组；如果零的个数等于$1$，那么$B[r]=P$，别的数都是$0$；如果$A$里没有$0$，则$B[i]=P/A[i]$。代码实现方面，可以直接in-place操作，而不需要再开一个数组。代码如下：

import java.util.Arrays;

public class Solution {
    /**
     * @param nums: an array of integers
     * @return: the product of all the elements of nums except nums[i].
     */
    public int[] productExceptSelf(int[] nums) {
        // write your code here
        int prod = 1, zeroCount = 0, zeroIdx = 0;
        for (int i = 0; i < nums.length; i++) {
            if (nums[i] != 0) {
                prod *= nums[i];
            } else {
                zeroCount++;
                zeroIdx = i;
            }
        }
        
        if (zeroCount >= 1) {
            Arrays.fill(nums, 0);
            if (zeroCount == 1) {
                nums[zeroIdx] = prod;
            }
        } else {
            for (int i = 0; i < nums.length; i++) {
                nums[i] = prod / nums[i];
            }
        }
        
        return nums;
    }
}

时间复杂度$O(n)$，空间$O(1)$。