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

拿到这个题，我也想了一会儿，是不是有什么位移或者是特别的符号能够代替除法，百度了一圈未果。

我只能按照自己也很鄙视的笨办法来做了。

不乘以A[i]，就是A[i]=1咯，然后就循环乘呗。

这是我的AC代码， 32 ms,639K

import java.util.ArrayList;
public class Solution {
public int[] multiply(int[] A) {
int [] B= new int[A.length];
int sum=1;
boolean flag=true;
int i=0;
while(flag){

int temp=A[i];

A[i]=1;
B[i]=1;
for(int j=0;j<A.length;j++){
B[i]*=A[j];
}
A[i]=temp;
i++;

if(i==B.length){
flag=false;
}
}

return B;

}
}

来看一下经典的做法：



这是一个三角形问题：

第一步：计算下三角，也就是每行的【】之前的乘积

第二步：计算上三角，也就是每行【】之后的乘积

大神的代码如下：（31ms 629K）

public class Solution {
    public int[] multiply(int[] A) {
        int length = A.length;
        int[] B = new int[length];
        if(length != 0 ){
            B[0] = 1;
            //计算下三角连乘
            for(int i = 1; i < length; i++){
                B[i] = B[i-1] * A[i-1];
            }
            int temp = 1;
            //计算上三角
            for(int j = length-2; j >= 0; j--){
                temp *= A[j+1];
                B[j] *= temp;
            }
        }
        return B;
    }
}

虽然只比我快1ms，但是一旦量级变大，我的代码还是不堪一击的。
当时在纸上递推的时候，我确实想过很多都是重复乘的问题，但是不知道该怎么去处理，因为乘的长度不一。

同时两个for循环， 还有就是分成两部分处理的思想，get√