构建乘积数组

Posted 2021-03-07 ustca

题目描述

给定一个数组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];）

思路

时间复杂度O(n)，空间复杂度O(1)。

代码

import java.util.ArrayList;
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 tmp = 1;
        for(int i = A.length - 2; i >= 0; i--) {
            tmp *= A[i+1];
            B[i] *= tmp;
        }
        return B;
    }
}

笔记

题名构建乘积数组，与矩阵图隐约有点联系。