剑指Offer:构建乘积数组(java版)

题目描述

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

循环解法

public class Solution {
    public int[] multiply(int[] A) {
        if(A==null || A.length<=1) return null;
        int len = A.length;
        int[] B = new int[len];
        B[0] = res(A,1,len-1);
        B[len-1] = res(A,0,len-2);
        for(int i =1; i+1<len; i++){
            B[i] = res(A,0,i-1)*res(A,i+1,len-1);
        }
        return B;
    }
    public int res(int[] arr,int L,int R){
        int res = 1;
        while(L<=R){
            res*=arr[L];
            L++;
        }
        return res;
    }
}

两次循环

假设数组长度为5，第一个循环,先从左往右算，得到每个数左边的所有乘积：
b[0] = 1;
b[1] = b[0] * a[0] = a[0]
b[2] = b[1] * a[1] = a[0] * a[1];
b[3] = b[2] * a[2] = a[0] * a[1] * a[2];
b[4] = b[3] * a[3] = a[0] * a[1] * a[2] * a[3];
第二个循环，从右往左算，得到每个数右边的所有乘积：
使用一个变量temp=1，代表右边的所有乘积
b[4] = b[4] * temp = a[0] * a[1] * a[2] * a[3]; temp = temp * a[4] = a[4];
b[3] = b[3] * temp = a[0] * a[1] * a[2] * a[4]; temp = temp * a[3] = a[4] * a[3];
b[2] = b[2] * temp = a[0] * a[1] * a[3] * a[4]; temp = temp * a[2] = a[4] * a[3] * a[2];
b[1] = b[1] * temp = a[0] * a[2] * a[3] * a[4]; temp = temp* a[1] = a[4] * a[3] * a[2] * a[1];
b[0] = b[0] * temp = a[4] * a[3] * a[2] * a[1];

public class Solution {
    public int[] multiply(int[] A) {
        if(A==null || A.length<=1) return null;
        int len = A.length;
        int[] B = new int[len];
        B[0] = 1 ;
        for(int i = 1 ;i<len;i++){
            B[i] = B[i-1] * A[i-1];
        }
        int temp = 1;
        for(int i = len-1;i>=0;i--){
            B[i]*=temp;
            temp*=A[i];
        }
        return B;
    }
}