剑指Offer_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[i],首先从前往后求解，先得到i前面的A[i]相乘结果，B[i] = B[i-1]×A[i-1]
同理从后开始，直到i+1，就可以得到B[i]。

实现

public class Solution {
    public int[] multiply(int[] A) {
        if (A == null || A.length == 0) return null;
        int[] B = new int[A.length];
        B[0] = 1;

        for (int i = 1; i < B.length; i++){
            B[i] = B[i-1] * A[i-1];
        }

        int num = A[A.length - 1];
        for (int i = B.length - 2; i >= 0; i--){
            B[i] *= num;
            num *= A[i];
        }

        return B;
    }
}