剑指之构建乘积数组

题目描述

给定一个数组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[i]=A[0]A[1]…*A[i - 1]A[i + 1]…*A[n - 1]分为两部分：
第一部分：A[0]A[1]…A[i - 1]
第二部分：A[i + 1]… A[n - 2] *A[n - 1]
即计算这两部分的乘积

public class Solution {
    public int[] multiply(int[] A) {
        if (A.length == 0)
        return A;
        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]; //计算B[i]=A[0]*A[1]*...*A[i-1]
        }
            int temp = 1;
        for (int j = A.length - 2; j >= 0; j--) {
            temp *= A[j + 1];   //从右到j左算B[i]*=A[i+1]*...*A[n-1]
            B[j] *= temp;
        }
        return B;
    }  
}