【数组】剑指Offer: 构建乘积数组

【在线编程】构建乘积数组

【问题描述】

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

【解题思路 & Java实现】

方法一：B[i]=A[0]A[1]…*A[i-1]A[i+1]…*A[n-1] = pre[i]*post[i]。其中 pre[i] = A[0]A[1]…A[i-1], post[i] = A[i+1]…*A[n-1].分别构造pre[i]和post[i]即可。

public class Solution {
    public int[] multiply(int[] A) {
        int[] B = new int[A.length];

        if (A == null || A.length == 0) {
            return B;
        }
        int[] pre = new int[A.length];
        int[] post = new int[A.length];

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

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

        for (int i = 0; i < B.length; i++) {
            B[i] = pre[i] * post[i];
        }
        return B;
    }
}

方法二：对方法一进行改进，不用开辟pre,post数组。

public class Solution {
    public int[] multiply(int[] A) {
        int[] B = new int[A.length];

        if (A == null || A.length == 0) {
            return B;
        }

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

        int temp = 1;
        for (int i = A.length - 2; i >= 0; i--) {
            temp = temp * A[i + 1];
            B[i] = B[i] * temp;
        }
        
        return B;
    }
}