构建乘积数组
题目
给定一个数组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]由上三角和下三角的矩阵乘积构成,因此,分别构建下三角矩阵C[i]和上三角矩阵D[i],二者相乘得到B[i]。
代码
public static int[] multiply(int[] A) {
if (A == null || A.length == 1) return null;
int length = A.length;
int[] B = new int[length];
int[] C = new int[length];
int[] D = new int[length];
int i = 0;
int j = length - 1;
C[i++] = 1;
D[j--] = 1;
for (; i < length && j >= 0 ; i++,j--) {
C[i] = C[i-1] * A[i-1];
D[j] = D[j+1] * A[j+1];
}
for (int k = 0; k < length; k++) {
B[k] = C[k] * D[k];
System.out.println("C["+k+"]="+C[k]+",D["+k+"]="+D[k]+",A["+k+"]="+A[k]);
}
return B;
}