构建乘积数组

Posted by DH on July 24, 2017

题目

给定一个数组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]就是下图每一行的乘积。

先计算下三角的乘积,再计算上三角。

代码

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