剑指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]。不能使用除法。
示例:
输入: [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]的下三角和上三角可分别通过递推公式求出来。
剑指offer-面试题66:构建乘积数组_第1张图片
于是得出如下步骤:

  • 初始化长度为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/

你可能感兴趣的:(数据结构和算法,算法,数据结构,leetcode,java,面试)