剑指Offer-构建乘积数组

题目描述

给定一个数组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]展开可以以对角线为界分成上下三角来计算。

B[i] = A[0]×A[1]×…×A[i-1]×A[i+1]×…×A[n-1]
从左到右算 B[i] = A[0]×A[1]×…×A[i-1]
从右到左算 B[i] ×= A[i+1]×…×A[n-1]

Java解题—上下三角

public class Solution {
    public int[] multiply(int[] A) {
        if(A==null)
            return null;
        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 i=A.length-2;i>=0;i--){
            temp *= A[i+1];
            B[i] *= temp;
        }

        return B;
    } 
}