剑指 Offer 66. 构建乘积数组【java详细题解】

剑指 Offer 66. 构建乘积数组

1. 题目

​ 给定一个数组 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

2. 思路

1. 遍历一遍内容数组得到乘机和，然后再遍历一遍内容数组用乘积和除以a[i]，时间复杂度O(n)，空间复杂度O(1)，但题目不可以用除法

2. 双循环遍历遇到下标相同的跳过，时间复杂度O(n^2)，空间复杂度O(n)，时间复杂度太高了，会超时，这里贴出代码

   class Solution {
       public int[] constructArr(int[] a) {
           int[] result = new int[a.length];
           Arrays.fill(result, 1);
           for (int i = 0; i < a.length; i++) {
               for (int j = 0; j < a.length; j++) {
                   if (i == j) continue;
                   result[i] *= a[j];
               }
           }
           return result;
       }
   }
   

3. 分别遍历以i为分界的数组的左右两边

可以把它理解为维护两个dp数组，一个dp数组计算矩阵的下三角，另一个dp数组维护矩阵的下三角，最后将两个dp数组相乘即可

在遍历下三角时候，将b[0]初始化为1，这样就可以有这样的一个状态转移方程求出左下角 b[i] = b[i - 1] * a[i - 1]

在遍历上三角时，需要一个和b[0] = 1一样作用的临时变量temp用来维护每层的乘积 tmp *= a[i + 1];

最后将两个dp数组乘起来，b[i] *= tmp;

时间复杂度O(n)，空间复杂度O(n)，也是本题的正解。

3. 代码

class Solution {
    public int[] constructArr(int[] a) {
        //特殊情况[]空数组
        if (a.length == 0) return new int[0];
        int[] b = new int[a.length];
        b[0] = 1;
        for(int i = 1; i < a.length; i++)
        b[i] = b[i - 1] * a[i - 1];
        int tmp = 1;
        for(int i = a.length - 2; i >= 0; i--) {
            tmp *= a[i + 1];
            b[i] *= tmp;
        }
        return b;
    }
}