剑指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]。不能使用除法。
 * 思考：剑指offer。 观察公式，B[i]公式中没有A[i]项，直观解法是连乘n-1个数字得到B[i]，但该方法时间复杂度为O(n^2)。
 * 更高效算法可以把B[i]看成A[0]~A[i-1]和A[i+1]~A[n-1]两部分连乘之后的乘积。
 * 
 * @author hexiaoli
 */

public class Main {
	public static int[] multiply(int[] A) {
		// 边界
		if (A == null || A.length <= 1) {
			return null;
		}
		int[] B = new int[A.length];

		B[0] = 1;
		// 计算左边A[0]~A[i-1]连乘部分(下三角连乘)
		for (int i = 1; i < A.length; i++) {
			B[i] = B[i - 1] * A[i - 1];
		}
		// 计算右边A[i+1]~A[n-1]连乘部分赋给temp，再乘上左半部分的连乘积
		int temp = 1;
		for (int j = A.length - 2; j >= 0; j--) {
			temp *= A[j + 1];
			B[j] *= temp;
		}		
		return B;
	}

	public static void main(String[] args) {
		int[] A = new int[] { 1, 2, 3, 4, 5 };
		int[] result = multiply(A);
		for (int i = 0; i < result.length; i++) {
			System.out.println(result[i]);
		}
	}
}