JZ51 构建乘积数组

JZ51 构建乘积数组

问题描述

给定一个数组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无意义，故而无法构建，因此该情况不会存在。

问题分析

根据题目描述，如果可以使用除法，就很简单。但是要求不能使用。
一种非常简单的方法就是连乘，而数组B中有n个值，每个值都要进行一次连乘，那么计算下来时间复杂度是O(n^2),效率很低。
所以要进行改进，首先我们来分析一下这个B数组形成过程:

B[0]	1	A[1]	A[2]	A[3]	…	A[n-2]	A[n-1]
B[1]	A[0]	1	A[2]	A[3]	…	A[n-2]	A[n-1]
B[2]	A[0]	A[1]	1	A[3]	…	A[n-2]	A[n-1]
B[3]	A[0]	A[1]	A[2]	1	…	A[n-2]	A[n-1]
B[n-2]	A[0]	A[1]	A[2]	A[3]	…	1	A[n-1]
B[n-1]	A[0]	A[1]	A[2]	A[3]	…	A[n-2]	1

右上方表格可以看出：
使A[i]=1
A[i]左边的乘积为：left[i]=A[0]xA[1]xA[2]x…xA[i-1]
A[i]右边的乘积为：right[i]=A[i+1]xA[i+2]x…xA[n-1]
left[i]=left[i-1]xA[i-1]
right[i]=right[i+1]xA[i+1]
B[i]=left[i]xright[i]
利用第一轮循环计算所有的左下三角，第二轮循环计算所有的右上三角。在第两轮for循环之时，进行对应的“组装”就可以得到对应的B[i]的值。

解法一(时间复杂度O(n^2))

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

解法二(时间复杂度O(n))

import java.util.ArrayList;
public class Solution {
    public int[] multiply(int[] A) {
    	//对于A长度为1的情况，B无意义 所以返回null
        if(A==null||A.length<2)
            return null;
        //创建B数组
        int[] B=new int[A.length];
        //计算左下三角的乘积
        B[0]=1;
        for(int i=1;i<A.length;i++)
            B[i]=B[i-1]*A[i-1];     
        //计算右上三角的乘积 和左下三角相乘 从最后一个开始乘
        int temp=1;
        for(int i=A.length-2;i>=0;i--){
        //right[i]=right[i+1]xA[i+1]
            temp*=A[i+1];
            B[i]*=temp;
        }
        return B;
    }
}