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

分析：

Bi是除了i对应元素之外的所有A元素的乘积。不能使用除法的话，最直观的解法是连乘，复杂度是O(n^2)，效率非常低。

剑指offer上面的解法，是把B[i]=A[0]*A[1]*...*A[i-1]*A[i+1]*...*A[n-1]分为两部分：C[i]=A[0]*A[1]*...*A[i-1]和D[i]=A[i+1]*...*A[n-1]。

总结来说就是C[i]=A[i-1]*C[i-1]，D[i]=A[i+1]*D[i+1].

思路

　　引用https://www.cnblogs.com/yongh/p/9971936.html。

　　考虑到计算每个B[i]时都会有重复，思考B[i]之间的联系，找出规律，提高效率。

如上图所示，可以发现：

　　　　B[i]的左半部分(红色部分)和B[i-1]有关（将B[i]的左半部分乘积看成C[i]，有C[i]=C[i-1]*A[i-1]），

　　　　B[i]的右半部分(紫色部分)与B[i+1]有关（将B[i]的右半部分乘积看成D[i]，有D[i]=D[i+1]*A[i+1]），

　　因此我们先从0到n-1遍历，计算每个B[i]的左半部分；  然后定义一个变量temp代表右半部分的乘积，从n-2到0遍历，令B[i]*=temp，而每次的temp与上次的temp关系，提前处理为temp*=A[i+1]。

具体代码如下：

import java.util.ArrayList;
public class Solution {
    public int[] multiply(int[] A) {
        int len=A.length;
        int[] B=new int[len];
        if(len==0){
            return B;
        }
        B[0]=1;//第一项不处理
        for(int i=1;i<len;i++){
            B[i]=B[i-1]*A[i-1];
        }
        int temp=1;//第n-1项不处理
        for(int i=len-2;i>=0;i--){
            temp*=A[i+1];
            B[i]*=temp;
        }
        return B;
    }
}