题目:
给你一个数组A[1..n]
,请你在O(n)
的时间里构造一个新的数组B[1..n]
,使得B[i]=A[1]*A[2]*...*A[n]/A[i]
。你不能使用除法运算。
解法1;
构建两个新的数组:C和D(为了方便解释,用了两个数组),
C[i] = A[0]*…*A[i-1] = C[i-1]*A[i-1]
D[i] = A[i+1]*…*A[n] = D[i+1]*A[i+1}
构建C和D都是O(n)
的时间复杂度(C从前到后遍历一遍数组,D从后到前遍历一边数组),然后,B[i] = C[i]*D[i]
也是O(n)
的时间复杂度。整体算法的时间复杂度是O(n)
。
解法2:
能够空间O(1)
的情况下实现
首先看看一个只有5个数的数组,A[1],A[2],A[3],A[4],A[5]
。
首先从头到尾遍历:
B[1] = A[1]
B[2] = B[1]*A[2]
B[3] = B[2]*A[3]
B[4] = B[3]*A[4]
B[5] = B[4], 临时变量 C=A[5]
然后从尾到头遍历:
B[4] = B[3]*C, C=C*A[4]
B[3] = B[2]*C, C=C*A[3]
B[2] = B[1]*C, C=C*A[2]
B[1] = C
通过这个小的例子,我们得到了算法,然后可以推广到任意多的元素。这个是面试中常用的技巧。
示例代码:
public class Mutil {
public static void main(String[] args) {
int[] ii = {1, 4, 3, 2, 6};
int[] res = new int[5];
int temp = ii[ii.length-1];
for(int i=0; i<ii.length; i++) {
if(i == 0) {
res[i] = ii[i];
} else if(i != ii.length-1) {
res[i] = res[i-1] * ii[i];
} else {
res[i] = ii[i];
}
}
for(int i=ii.length-1; i>=0; i--){
if(i == ii.length-1) {
res[i] = res[i-1];
} else if( i == 0) {
res[i] = temp;
} else {
res[i] = res[i-1] * temp;
temp = temp * ii[i];
}
}
for(int i: res) {
System.out.println(i);
}
}
}