剑指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]。不能使用除法。（注意：规定B[0] = A[1] * A[2] * … * A[n-1]，B[n-1] = A[0] * A[1] * … * A[n-2];）

实现

思路：题目要求B的i个元素等于A中除了i个元素所以元素乘积

下三角用连乘可以很容求得，上三角，从下向上也是连乘。

/**
思路：题目要求B的i个元素等于A中除了i个元素所以元素乘积
**/
import java.util.ArrayList;
public class Solution {
    public int[] multiply(int[] A) {
        if(A==null) return null;
        if(A.length==0) return new int[0]; 
        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];
        }
        //B[n]就会在第一个循环的时候，计算好，而B[0]则在下一个循环的时候计算好。
        //上三角从下往上连乘
        int temp=1;
        for(int j=len-2;j>=0;j--){
            temp*=A[j+1];
            B[j]*=temp;
        }
        return B;
    }
}