[LeetCode]（面试题66）构建乘积数组

题目

给定一个数组 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]。不能使用除法。

示例:

输入: [1,2,3,4,5]
输出: [120,60,40,30,24]

提示：

• 所有元素乘积之和不会溢出 32 位整数
• a.length <= 100000

解题思路

构建两个dp数组，记为left，right，分别维护 i 左侧、右侧元素的连乘积。

• 1）计算每个元素左边所有元素的乘积，结果记录在left数组中，初始化时 left[0] = 1，因为第一个位置左边没有元素；
• 2）计算每个元素右边所有元素的乘积，结果记录在right数组中，同理 right[n-1] = 1;
• 3）最终，每个位置除当前元素外，其它元素的乘积 left[i] = left[i] * right[i]。left 是最终的结果，为了节省空间，先用 left 记录左边元素的乘积。

复杂度分析：
时间复杂度：O(n)。n 为数组长度，其中需要遍历两遍数组 a，遍历一遍数组 left，各使用 O(n) 时间。
空间复杂度：O(n^2)。

代码

class Solution {
    public int[] constructArr(int[] a) {
        int n = a.length;
        if(n==0){
            return new int[0];
        }
        // left[i] 表示第i个元素左边的连乘积
        int[] left = new int[n];
        // right[i] 表示第i个元素右边的连乘积        
        int[] right = new int[n];
        left[0] = 1;
        right[n-1] = 1;
        for(int i=1; i<n; i++){
            left[i] = left[i-1]*a[i-1];
        }
        for(int i=n-2; i>=0; i--){
            right[i] = right[i+1]*a[i+1];
        }
        for(int i=0; i<n; i++){
            left[i] *= right[i];
        }
        return left;
    }
}