JZ-051-構建乘積陣列

雄獅虎豹發表於2022-01-15
陣列

構建乘積陣列

題目描述

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

  • （注意：規定B[0] = A[1] A[2] ... A[n-1]，B[n-1] = A[0] A[1] ... A[n-2];）
  • 對於A長度為1的情況，B無意義，故而無法構建，因此該情況不會存在。

題目連結: 構建乘積陣列

程式碼

/**
 * 標題：構建乘積陣列
 * 題目描述
 * 給定一個陣列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]。不能使用除法。
 * （注意：規定B[0] = A[1] * A[2] * ... * A[n-1]，B[n-1] = A[0] * A[1] * ... * A[n-2];）
 * 對於A長度為1的情況，B無意義，故而無法構建，因此該情況不會存在。
 * 題目連結：
 * https://www.nowcoder.com/practice/94a4d381a68b47b7a8bed86f2975db46?tpId=13&&tqId=11204&rp=1&ru=/ta/coding-interviews&qru=/ta/coding-interviews/question-ranking
 */
public class Jz51 {

    public int[] multiply(int[] A) {
        int n = A.length;
        int[] B = new int[n];
        for (int i = 0, product = 1; i < n; product *= A[i], i++) {       /* 從左往右累乘 */
            B[i] = product;
        }
        for (int i = n - 1, product = 1; i >= 0; product *= A[i], i--) {  /* 從右往左累乘 */
            B[i] *= product;
        }
        return B;
    }

    public static void main(String[] args) {
        Jz51 jz51 = new Jz51();
        int[] A = new int[]{1, 2, 3, 4, 5, 6};
        int[] result = jz51.multiply(A);
        for (int num : result) {
            System.out.print(num + " ");
        }
    }
}
【每日寄語】 今天又是升級打怪的一天，加油。

相關文章