题目描述
给定一个数组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];)
给定代码
public class Solution {
public int[] multiply(int[] A) {
}
}
题解一
第一次想出来的办法,左右打表,左表为chartLeft,右表为chartRight。
chartLeft[i]
代表A[0] * A[1] * ... * A[i]
chartRight[i]
代表A[n - 1] * A[n - 2] * ... * A[i]
通过两个循环构造左右表,然后再通过一个循环给结果集填充数值。
代码如下:
public int[] multiply(int[] A) {
int [] arrayB = new int[A.length];
// 边界值判断
if (A.length <= 1) {
return arrayB;
}
// 左右打表
// chartLeft[i] = A[0] * A[1] * ... * A[i]
// chartRight[i] = A[n - 1] * A[n - 2] * ... * A[i]
int [] chartLeft = new int[A.length];
int [] chartRight = new int[A.length];
// temp 保存A[0] * A[1] * ... * A[i - 1]的缓存值
int temp = 1;
for (int i = 0 ; i < A.length ; i++)
{
temp *= A[i];
chartLeft[i] = temp;
}
// temp 保存A[n - 1] * A[n - 2] * ... * A[i + 1]的缓存值
temp = 1;
for (int i = A.length - 1 ; i >= 0 ; i--)
{
temp *= A[i];
chartRight[i] = temp;
}
// 结果集填充数值
arrayB[0] = chartRight[1];
arrayB[A.length - 1] = chartLeft[A.length - 2];
for (int i = 1 ; i < A.length - 1 ; i++)
{
arrayB[i] = chartLeft[i - 1] * chartRight[i + 1];
}
return arrayB;
}
题解二
评论区的思路。
既然B[i] = A[0]*A[1]*...*A[i-1]*A[i+1]*...*A[n-1]
,那么可以分两步,
- 左累乘:B[i] = A[0] * A[1] * … * A[i - 1]
- 右累乘:B[i] *= A[n - 1] * A[n - 2] * … * A[i + 1]
所以只需要从0开始往后循环
给每个B[i]赋上前一半的值,再从n-1开始往前循环
累乘后一半的值即可。
代码如下:
public int[] multiply1(int[] A) {
int [] arrayB = new int[A.length];
// 边界值判断
if (A.length <= 1) {
return arrayB;
}
int temp = 1;
// 此循环结束后,每个B[i]都已经完成了A[0] * A[1] * ... * A[i - 1]
for (int i = 0 ; i < A.length ; i++)
{
arrayB[i] = temp;
temp *= A[i];
}
temp = 1;
// 每个B[i]在上轮基础上乘了右边的A[n - 1] * A[n - 2] * ... * A[i + 1]
for (int i = A.length - 1 ; i >= 0 ; i--)
{
arrayB[i] *= temp;
temp *= A[i];
}
return arrayB;
}