剑指 Offer 66. 构建乘积数组java题解

N刷发现什么都不会。
这题我也不会

class Solution {
    public int[] constructArr(int[] a) {
        if(a.length == 0) return new int[0];
        int[] b = new int[a.length];
        b[0] = 1;
        int temp = 1;

        // 计算b：b[i]表示a[0...i-1]的乘积
        for(int i=1; i<a.length; i++){
            b[i] = b[i-1] * a[i-1];
        }
        // temp 表示a[i...n-1]的乘积
        for(int i=a.length-1; i>=0; i--){
            b[i] *= temp;
            temp *= a[i];
        }
        return b;
    }
}

1.题目

https://leetcode-cn.com/problems/gou-jian-cheng-ji-shu-zu-lcof

B[i]的值是A中除了A[i]以外所有元素的乘积。

2.分析

本题难点在于不让用除法。
整体思路，结果集中任何一个元素 = 其左边所有元素的乘积 * 其右边所有元素的乘积。一轮循环构建左边的乘积并保存在结果集中，二轮循环构建右边乘积的过程，乘以左边的乘积，并将最终结果保存。

2.1 示例

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;
        int tmp=1;//辅助变量
        //计算下三角乘积
        //i从第2个开始
        for(int i=1;i<a.length;i++){
            b[i]=b[i-1]*a[i-1];
        }
        //计算上三角乘积
        //i从倒数第2个开始
        for(int i=a.length-2;i>=0;i--){
            tmp=tmp*a[i+1];
            b[i]=b[i]*tmp;
        }
        return b;
    }
}

4.复杂度

时间复杂度 O(N)： 其中 N为数组长度，两轮遍历数组 a ，使用 O(N) 时间。
空间复杂度 O(1)： 变量 tmp使用常数大小额外空间（数组 b作为返回值，不计入复杂度考虑）。

5.结果

空间复杂度O(N)，构建左右数组

class Solution {
    public int[] productExceptSelf(int[] nums) {
        int len=nums.length;
        int[] res=new int[len];
        Arrays.fill(res,1);
        int[] start=new int[len];
        int[] end=new int[len];
        start[0]=nums[0];
        end[len-1]=nums[len-1];
        for(int i=1;i<len;i++){
            start[i]=start[i-1]*nums[i];
            end[len-1-i]=end[len-1-i+1]*nums[len-1-i];
        }
        for(int i=0;i<len;i++){
            if(i-1>=0){
                res[i]=start[i-1];
            }
            if(i+1<len){
                res[i]*=end[i+1];
            }
            //res[i]=start[i-1]*end[i+1];
        }
        return res;
    }
}

左右数组只保留左数组，作为输出数组。所以没有额外的空间复杂度。

class Solution {
    public int[] productExceptSelf(int[] nums) {
        int len=nums.length;
        int[] start=new int[len];
        //start[i]:i左边的所有乘积
        Arrays.fill(start,1);
        for(int i=1;i<len;i++){
            start[i]=start[i-1]*nums[i-1];
        }
        int last=nums[len-1];
        for(int i=len-2;i>=0;i--){
            start[i]=start[i]*last;
            last=last*nums[i];
            //需要包含所有右边的乘积，所以计算下一个结果时需要将当前数字算上
        }
        return start;
    }
}

超出内存限制的代码

class Solution {
    public int[] constructArr(int[] a) {
        int n=a.length;
        int[][] dp=new int[n][n];
        for(int i=0;i<n;i++){
            dp[i][i]=a[i];
        }
        
        for(int i=0;i<n-1;i++){
            for(int j=i+1;j<n;j++){
                dp[i][j]=dp[i][j-1]*a[j];
            }
        }
        int[] res=new int[n];
        for(int i=0;i<n;i++){
            if(i==0)
                res[i]=dp[i+1][n-1];
            else if(i==n-1)
                res[i]=dp[0][n-2];
            else
                res[i]=dp[0][i-1]*dp[i+1][n-1];
        }
        return res;
    }
}