构建乘积数组

题目

给定一个数组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数组先从 1计算到n - 1，然后再从n - 2计算到0，这样B数组就是一个答案了。具体代码如下：

package com.zhumq.leetcode;
import java.util.Arrays;
import java.util.Scanner;
import org.junit.Test;

public class MultiplyArray {

    public int[] multiplyArray(int A[],int n) {
        //首先B[i]=A[1]*A[2]*A[3]*...*A[i-1] 
        int[] B = new int[n];
        B[0] = 1;
        for(int i = 1;i<n;i++) {
            //利用前面求过的值
            B[i] = B[i-1] * A[i-1];
        }

        /*
         * 然后乘上后面的A[i+1]*A[i+2]*...*A[n-1]
         * pre保存前面求过的A[i+1]*...*A[n-1]
         */
        int pre = 1;
        for(int i = n-2;i>=0;i--) {
            pre *= A[i+1];
            /*
             * B[n-1]=A[1]*...*A[n-2]所以前已经求出， 这里不需要再求B[n-1]
             * 相当于用上个for求的B[i]再乘上后面的一部分
             */
            B[i] = B[i]*pre;
        }
        return B;
    }

    @Test
    public void test1() {
        Scanner sc = new Scanner(System.in);
        while(sc.hasNext()) {
            int n = sc.nextInt();
            int[] arr = new int[n];
            for(int i = 0;i<n;i++) {
                arr[i] = sc.nextInt();
            }
            System.out.println(Arrays.toString(multiplyArray(arr, n)));
        }
        sc.close();
    }
}