题目:给定一个数组 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]位置的元素对应的a[i] =1;因此我们的第一种思路就是先把a[i]==1;然后遍历进行乘积。但是这样跑出来以后,测试案例最后一个会通不过,时间超时。
因此,我们换一个思路,上面这个思路,出现了重复计算没有利用之前计算的信息。我们可以先计算下三角的值,然后在计算上三角的值。
具体代码:
class Solution {
public int[] constructArr(int[] a) {
//按照题意可以知道,数组b的对角线是1,如果只将对应的元素换成1,在进行遍历求积的话会导致时间通不过
//因此,可以分别求上下三角,最后在乘在一起,减少计算量
int[] b = new int[a.length];
if(a.length == 0) return new int[0];
b[0] = 1;
for(int i=1;i<a.length;i++){
b[i] = b[i-1]*a[i-1];
}
int tmp = 1;
for(int i=a.length-2;i>=0;i--){
tmp *= a[i+1];
b[i] *= tmp;
}
return b;
}
}