上三角下三角建構乘積數組

題目描述：

給定一個數組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]。不能使用除法。

解決思路：

• 方法：先算下三角，然後算上三角，最後相乘即可

  

  上下三角.jpg

public class Solution {
    public int[] multiply(int[] A) {
        /* 時間複雜度O(n^2)
        int[] B = new int[A.length];
        for(int i = 0; i < B.length; i++){
            int sum = 1;
            for(int j = 0; j < A.length; j++){
                if(i == j)
                    continue;
                sum *= A[j];
            }
            B[i] = sum;
        }
        
        return B;
        */
        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 temp = 1;
        for(int i = A.length - 2; i >= 0; i--){
            temp *= A[i + 1];
            B[i] *= temp;
        }
        return B;
    }
}