2021.10.25 - JZ66.构建乘积数组

1. 题目

2. 思路

(1) 乘积数组

• 根据题目要求，B中的第i个元素即为A中下标为[0,i-1]的元素的乘积再乘以下标为[i+1,n-1]的元素的乘积。
• 创建乘积数组left和right，left[i]表示A中下标为[0,i]的元素的乘积，right[i]表示A中下标为[i,n-1]的元素的乘积，遍历两次A数组即可。
• 再遍历一次A数组，根据B[i]=left[i-1]*right[i+1]求出B数组，左右两端的元素单独处理。

(2) 乘积数组优化

• 与(1)的思路基本相同，先从左往右遍历一次A数组，求出第i个元素左边的元素的乘积，再从右往左遍历一次A数组，求出第i个元素右边的元素的乘积，两次结果累乘即为最终结果。

3. 代码

public class Test {
    public static void main(String[] args) {
    }
}

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

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