【在线编程】构建乘积数组
【问题描述】
给定一个数组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]。不能使用除法。
【解题思路 & Java实现】
方法一:B[i]=A[0]A[1]…*A[i-1]A[i+1]…*A[n-1] = pre[i]*post[i]。其中 pre[i] = A[0]A[1]…A[i-1], post[i] = A[i+1]…*A[n-1].分别构造pre[i]和post[i]即可。
public class Solution {
public int[] multiply(int[] A) {
int[] B = new int[A.length];
if (A == null || A.length == 0) {
return B;
}
int[] pre = new int[A.length];
int[] post = new int[A.length];
pre[0] = 1;
for (int i = 1; i < A.length; i++) {
pre[i] = pre[i-1] * A[i-1];
}
post[post.length-1] = 1;
for (int i = A.length-2; i >= 0; i--) {
post[i] = post[i+1] * A[i+1];
}
for (int i = 0; i < B.length; i++) {
B[i] = pre[i] * post[i];
}
return B;
}
}
方法二:对方法一进行改进,不用开辟pre,post数组。
public class Solution {
public int[] multiply(int[] A) {
int[] B = new int[A.length];
if (A == null || A.length == 0) {
return B;
}
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 = temp * A[i + 1];
B[i] = B[i] * temp;
}
return B;
}
}