构建乘积数组

题目描述

给定一个数组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]。不能使用除法。

解法

1.A[i]置为1法，复杂度为O(n^2)

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

2.左右法,复杂度为O(n),先将A[i]的左右的分别乘积预先计算好，然后统一计算数组B，用上了动态规划的思想。

import java.util.ArrayList;
public class Solution {
    public int[] multiply(int[] A) {
        if(A==null||A.length==0)
            return A;
        int[] left = new int[A.length];//记录除了自己，左边的乘积
        int[] right = new int[A.length];//记录除了自己，右边的乘积
        right[A.length-1] = 1;
        for(int i = A.length-2;i>=0;i--){
            right[i] = right[i+1]*A[i+1];
        }
        left[0] = 1;
        for(int i = 1;i<A.length;i++){
            left[i] = left[i-1]*A[i-1];
        }
        //可以更简便
        int[] B = new int[A.length];
        for(int i = 0;i<A.length;i++){
            B[i] = left[i]*right[i];
        }
        return B;
    }
}