【Lintcode】50. Product of Array Exclude Itself

题目地址：

https://www.lintcode.com/problem/product-of-array-exclude-itself/description

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

如果不许用除法的话，可以参考https://blog.csdn.net/qq_46105170/article/details/108544550。

否则的话，思路是，用三个变量分别保存非零数乘积（记为$p$）、零的下标、零的个数。如果零的个数大于$1$，那就返回$0$数组；如果零的个数等于$1$，那零的那个位置要填非零数乘积，别的位置填$0$；如果没有$0$，那$B[i]=\frac{p}{A[i]}$。代码如下：

import java.util.ArrayList;
import java.util.Collections;
import java.util.List;

public class Solution {
    /*
     * @param nums: Given an integers array A
     * @return: A long long array B and B[i]= A[0] * ... * A[i-1] * A[i+1] * ... * A[n-1]
     */
    public List<Long> productExcludeItself(List<Integer> nums) {
        // write your code here
        List<Long> res = new ArrayList<>();
        for (int i = 0; i < nums.size(); i++) {
            res.add(0L);
        }
        
        int zeroIdx = -1, zeroCount = 0;
        long product = 1;
        for (int i = 0; i < nums.size(); i++) {
            if (nums.get(i) != 0) {
                product *= nums.get(i);
            } else {
                zeroCount++;
                zeroIdx = i;
            }
        }
        
        if (zeroCount > 1) {
            Collections.fill(res, 0L);
        } else if (zeroCount == 1) {
            for (int i = 0; i < nums.size(); i++) {
                if (i != zeroIdx) {
                    res.set(i, 0L);
                } else {
                    res.set(i, product);
                }
            }
        } else {
            for (int i = 0; i < nums.size(); i++) {
                res.set(i, product / nums.get(i));
            }
        }
        
        return res;
    }
}

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