剑指 Offer 66. 构建乘积数组

剑指 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]。不能使用除法。

解答

class Solution {
    /**
    根据数组A构建前缀乘积数组C1（不含自身，即C[i] = C[0]*C[1]*...C[i-1]）
    根据数组A构建后缀乘积数组C2（不含自身，即C[i] = C[i+1]*C[i+2]*...C[length-1]）

    B[i] = C1[i] * C2[i]
     */
    public int[] constructArr(int[] a) {
        if(a.length == 0){
            return new int[0];
        }
        int[] C1 = new int[a.length];
        int[] C2 = new int[a.length];

        C1[0] = 1;
        for(int i = 1;i < C1.length;i++){
            C1[i] = C1[i-1] * a[i-1];
        }

        C2[C2.length - 1] = 1; 
        for(int i = C2.length - 2;i >= 0;i--){
            C2[i] = C2[i+1] * a[i+1];
        }

        int[] B = new int[a.length];
        for(int i = 0;i < a.length;i++){
            B[i] = C1[i] * C2[i];
        }

        return B;
    }
}