题目描述
给定一个数组 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]
提示:
所有元素乘积之和不会溢出 32 位整数
a.length <= 100000
方法一(分组递推)
1.解题思路
B[i]=A[0]×A[1]×…×A[i-1]×A[i+1]×…×A[n-1],根据B[i]的表达式,可以把每一项以表格的形式列举出来,B[i]的下三角和上三角可分别通过递推公式求出来。
于是得出如下步骤:
- 初始化长度为n的数组int[] b
- 根据递推公式b[i+1]=b[i]*a[i],得到b[i]的下三角部分
- 根据递推公式b[n-i]=a[n]*a[n-1]*……*a[n-i+1],得到b[i]的上三角部分
- 将上三角和下三角部分合并,即可得到整个b数组
2.代码实现
class Solution {
public int[] constructArr(int[] a) {
if(a==null||a.length==0) return new int[0];
int n=a.length;
int[] b=new int[n];
b[0]=1;
for(int i=0;i<n-1;i++){
b[i+1]=b[i]*a[i];
}
int temp=1;
for(int i=n-2;i>=0;i--){
temp*=a[i+1];
b[i]*=temp;
}
return b;
}
}
3.复杂度分析
- 时间复杂度:线性遍历a数组两次,所以时间复杂度为O(n)。
- 空间复杂度:需要额外常数级别的内存空间,所以空间复杂度为O(1)。
剑指offer全集入口: 请戳这里
参考资料:
https://leetcode-cn.com/problems/gou-jian-cheng-ji-shu-zu-lcof/solution/mian-shi-ti-66-gou-jian-cheng-ji-shu-zu-biao-ge-fe/