剑指 Offer.66-构建乘积数组
剑指 Offer.66-构建乘积数组
题目链接
<剑指 Offer 66. 构建乘积数组>
问题描述
给定一个数组 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]。不能使用除法。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/gou-jian-cheng-ji-shu-zu-lcof
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
个人想法(双循环求解)
由于该题规定不能使用除法,所以通过双循环的方式进行求解,在通过最后一个测试用例的时候超时
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20
| class Solution { public int[] constructArr(int[] a) { int[] res = new int[a.length]; for(int i=0;i<a.length;i++){ int subRes = 1; for(int j=0;j<a.length;j++){ if(j == i){ continue; }else{ subRes*=a[j]; } } res[i] = subRes; } return res; } }
|
官方解法(记忆化搜索)
将整个数组放入一个表格中,主对角线上的元素都为1
,计算左右三角矩阵每行的乘积,最后再将左右侧三角矩阵的乘积直接乘起来,得到的就是除第i
个元素的乘积,在思想上类似动态规划。
代码
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24
| class Solution { public int[] constructArr(int[] a) { if(a == null || a.length == 0) return a; int len = a.length; int [] left = new int[len]; int [] right = new int[len];
left[0] = right[len-1] = 1; for(int i=1;i<len;i++){ left[i] = left[i-1] * a[i-1]; }
for(int j = len-2;j>=0;j--){ right[j] = right[j+1] * a[j+1]; }
int[] ans = new int[len]; for(int i=0;i<len;i++){ ans[i] = left[i] * right[i]; }
return ans; } }
|