剑指Offer49--面试题66. 构建乘积数组

题目描述

给定一个数组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];）

暴力搜索

import java.util.ArrayList;
public class Solution {
    public int[] multiply(int[] A) {
        int len = A.length;
        int[] res = new int[len];
        int i = 0;
        while(i<len){
            int a = 1;
            for(int k=0; k<len; k++){
                if(k!=i){
                    a *= A[k];
                }
            }
            res[i] = a;
            i++;
        }
        
        return res;
    }
}

先算左边，再算右边， 将 时间复杂度降低为O(n)

class Solution {
    public int[] constructArr(int[] A) {
         int len = A.length;
        if(len==0) return new int[0];
        int[] res = new int[len];
        res[0] = 1;
        // 算出 i前各个数组的乘积
        for( int i = 1;i<len; i++){
            res[i] = res[i-1] * A[i-1];
        }
        int temp = 1;
        for(int i=len-2;i>=0; i--){
           temp *= A[i+1]; 
            res[i] *= temp;
        }
        
        
        return res;
    }
}