【剑指offer】51.构建乘积数组

51.构建乘积数组

知识点：数组；

题目描述

给定一个数组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[i],就是把A中数组元素全部乘一次，然后除以A[i]，但是题目中不能用除法，所以可以每次先将A[i]的值赋为1，然后再去相乘；

public class Solution {
    public int[] multiply(int[] A) {
        int[] B = new int[A.length];
        int temp;
        for(int i = 0; i < A.length; i++){
            int count=1;
            temp = A[i];  //临时把A[i]的值存起来，这一轮结束后再还给A[i];
            A[i] = 1;
            for(int j = 0; j < A.length; j++){
                count *= A[j];
            }
            A[i] = temp;
            B[i] = count;
        }
        return B;
    }
}

这样去解用了for的嵌套，时间复杂度为O(n^2);其实可以发现我们每次都要去将这么多元素的值去乘一遍，每次只有一个元素不一样，这样太浪费了，所以有了下面这样的解法。

解法二：

经过上述分析后可以得到下图，其实就是将求B中第i个元素的值，那就是让A中的第i个元素值为1

算每行的Bi就是将这行所有的元素乘一下就可以了，可以发现在左下角，在算B[i的时候，其实是要用到B[i-1]的，它们的关系是B[i]=B[i-1]*A[i-1];再看右上角，从下往上，假设值为temp吧，那temp[i]=temp[i+1] *A[i+1];然后再把左边和右边相乘就可以了。

public class Solution {
    public int[] multiply(int[] A) {
        int[] B = new int[A.length];
        B[0] = 1;
        int temp = 1;
        for(int i = 1; i < A.length; i++){
            B[i] = B[i-1] * A[i-1]; //计算左下角；
        }
        for(int j = A.length-2; j >= 0; j--){
            temp *= A[j+1];  //计算右下角和整个B；
            B[j] *= temp;
        }
        return B;
    }
}

这样利用到了先前的信息，减少了浪费，时间复杂度为O(n);