剑指51 - 构建乘积数组

剑指51 - 构建乘积数组

给定一个数组A[0,1,...,n-1],请构建一个数组B[0,1,...,n-1],其中B中的元素B[i]=A[0]*A[1]*...*A[i-1]*A[i+1]*...*A[n-1]。不能使用除法。（注意：规定B[0] = A[1] * A[2] * ... * A[n-1]，B[n-1] = A[0] * A[1] * ... * A[n-2];）
对于A长度为1的情况，B无意义，故而无法构建，因此该情况不会存在。
示例1
输入
[1,2,3,4,5]
输出
[120,60,40,30,24]

解法一：暴力求解，算法复杂度 n^2

import java.util.ArrayList;
public class Solution {
    public int[] multiply(int[] A) {
        if(A == null || A.length <=1) return new int[0];
        int [] result = new int[A.length];
        result[0] = 1;
        for(int i=1; i<A.length; i++) {
            if(result[0] == 0) break;
            result[0] *=  A[i];
        }
        for(int i=1; i<result.length; i++) {
            if(A[i] == 0) {
                result[i] = getExceptZero(A, i);
                continue;
            }
            result[i] = result[0]/A[i] * A[0];
        }
        return result;
    }
    
    private int getExceptZero(int[] arr, int i) {
        // i位置是0
        int ret = 1;
        for(int j=0; j<arr.length; j++) {
            if(j == i) continue;
            if(ret == 0) return 0;
            ret *=  arr[j];
        }
        return ret;
    }
}

解法二：

    下三角计算公式：B[i] = B[i-1] + A[i-1];
    上三角借助中间变量temp实现对B中元素的另一半补充

import java.util.ArrayList;
public class Solution {
    public int[] multiply(int[] A) {
        if(A == null || A.length <=1) return new int[0];
        int [] B = new int[A.length];
        B[0] = 1;
        // 下三角
        for(int i=1; i<A.length; i++) {
            B[i] = B[i-1] * A[i-1];
        }
        // 上三角倒着推
        int temp = 1;
        for(int j=A.length-2; j>=0; j--) {
            temp *= A[j+1];
            B[j] *= temp;
        }
        return B;
    }
}