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

解题思路

第一步：创建一个数组B
第二步：循环一次，把B[i]赋值为 A[0] * A[1] * ··· * A[i-1];
第三步：从后往前循环一次，把B[i]赋值为B[i] * A[n] * A[n-1] * ··· A[i+1]。

代码如下

import java.util.*;
public class Solution {
    public int[] multiply(int[] A) {
        if (A.length == 0)
            return null;
        int[] B = new int[A.length];
        Arrays.fill(B,1);
        for (int i = 1; i < B.length; i++) {
            B[i] = B[i-1] * A[i-1];
        }
        int temp = 1;
        for (int i = B.length-1; i >= 0; i--) {
            B[i] = B[i] * temp;
            temp = temp * A[i];
        }
        return B;
    }
}

提交结果