给定一个数组 A[0,1,…,n-1],请构建一个数组 B[0,1,…,n-1],其中 B[i] 的值是数组 A 中除了下标 i 以外的元素的积, 即 B[i]=A[0]×A[1]×…×A[i-1]×A[i+1]×…×A[n-1]。不能使用除法。
示例:
输入: [1,2,3,4,5]
输出: [120,60,40,30,24]
方法1:建立两个数组分别放该点左侧和右侧的乘积,求一个数的乘积时,只需要求前一个位置左数组的值,再乘后一个位置右数组的值即可。
class Solution {
public int[] constructArr(int[] a) {
if(a==null || a.length ==0){
return new int[0];
}
int len = a.length;
int[] res = new int[len];
int[] left = new int[len];
int[] right = new int[len];
left[0] = a[0];
right[len-1] = a[len-1];
for(int i=1;i<len;i++){
left[i] = left[i-1] * a[i];
}
for(int i=len-2;i>=0;i--){
right[i] = right[i+1] * a[i];
}
res[0] = right[1];
res[len-1] = left[len-2];
for(int i=1;i<len-1;i++){
res[i] = left[i-1] * right[i+1];
}
return res;
}
}
方法2:思想其实是一样的,就是初始化的时候不同,这种方法可以进行优化
class Solution {
public int[] constructArr(int[] a) {
if(a==null || a.length ==0){
return new int[0];
}
int len = a.length;
int[] res = new int[len];
int[] left = new int[len];
int[] right = new int[len];
left[0] = 1;
right[len-1] = 1;
for(int i=1;i<len;i++){
left[i] = left[i-1] * a[i-1];
}
for(int i=len-2;i>=0;i--){
right[i] = right[i+1] * a[i+1];
}
for(int i=0;i<len;i++){
res[i] = left[i] * right[i];
}
return res;
}
}
优化代码:
class Solution {
public int[] constructArr(int[] a) {
//边界条件的判断
if (a == null || a.length == 0)
return a;
int length = a.length;
int[] res = new int[length];
res[0] = 1;
//当前元素左边的所有元素乘积(不包含当前元素)
for(int i=1;i<length;i++) {
res[i] = res[i-1] * a[i-1];
}
int right = 1;
//right表示当前元素右边所有元素的乘积(不包含当前元素),
//res[i]表示的是左边的乘积,他俩相乘就是
//除了自己以外数组的乘积
for(int i=length-1;i>=0;i--) {
res[i] *= right;
right *= a[i];
}
return res;
}
}
感觉不是很好理解。。但是速度提升了一倍。