描述
给定一个数组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[0] = A[1] * A[2] * … * A[n-1],B[n-1] = A[0] * A[1] * … * A[n-2];)
对于A长度为1的情况,B无意义,故而无法构建,因此该情况不会存在。
示例1
输入: [1,2,3,4,5]
返回值: [120,60,40,30,24]
思路
由于
left[i] = A[0]*…*A[i-1]
right[i] = A[i+1]*…*A[n-1]
所以:
B[i] = left[i] * right[i]
可知:
left[i+1] = A[0]*…A[i-1]*A[i]
right[i+1] = A{i+2]*…*A[n-1]
于是,
left[i+1] = left[i] * A[i]
right[i] = right[i+1] * A[i+1]
所以,我们可以先把所有的left[i]求出,right[i]求出。
可以用数组保存中间变量left[i] 和 right[i],但是空间复杂度高
本次代码用临时变量做优化,简化空间复杂度
代码
import java.util.ArrayList;
public class Solution {
public int[] multiply(int[] A) {
if(A == null || A.length < 2) {
return null;
}
int[] B = new int[A.length];
for (int i=0; i<B.length; ++i) {
//初始化数组
B[i] = 1;
}
for (int i=1; i<A.length; i++) {
// left[i]用B[i]代替
B[i] = B[i-1] * A[i-1];
}
int temp = 1;
for (int j=A.length-2; j>=0; j--) {
// right[i]用tmp代替
temp *= A[j+1];
B[j] *= temp;
}
return B;
}
}