剑指 Offer 66. 构建乘积数组
按照一般的思路就是将所有的相乘,然后除以每一位数字就是答案,但是题目要求我们不能使用除法,因此我们会想到每次遍历到每个数字的时候,在遍历一遍数组,将除开自己以外的数字相乘,但是这样做的时间复杂度确是\(O(N^2)\),导致超时,因此我们需要想另外一种方法来解决
根据题意,我们可以知道B[i]=A[0]*A[1]*A[2]*...*A[i-1]*A[i+1]*...*A[n-1]
,所以我们以i
为分界线,将这个拆成两部分,所以B[i]
就等于A[0]*...*A[i-1]
与A[i+1]*...*A[n-1]
的乘积,所以数组B可以看作用一个矩阵来创建。我们以[1, 2, 3, 4]
为例,如图所示:
可以看出,对角线是都为1。然后从第二行开始,我们先计算下三角的每一行的乘积:B[i] = B[i-1] * A[i-1]
从倒数第二行开始,再从下往上计算上结果:B[i] *= A[i+1]
,因为左边部分已经计算出来了,所以直接拿来乘就可以了
class Solution { public int[] constructArr(int[] a) { int length = a.length; if (length == 0) { return new int[0]; } int[] arr = new int[length]; // 先构建左下三角形 arr[0] = 1; for (int i = 1; i < length; i++) { arr[i] = arr[i-1] * a[i-1]; } // 构建右上三角形同时计算结果 int temp = 1; // 从倒数第二个开始往前 for (int i = length-2; i >= 0; i--) { temp *= a[i+1]; arr[i] *= temp; } return arr; } }