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

方法一

如果能使用除法，可以求出A[0]*...*A[n-1]的乘积结果记为product，则每个B[i] = product / A[i]。

但是不能使用除法，比较简单的办法就是按照公式B[i]=A[0]*A[1]*...*A[i-1]*A[i+1]*...*A[n-1]，直接for循环求一遍，但是这种方法时间复杂度为O(n^2)，不是很好。

代码

class Solution {
public:
    vector<int> multiply(const vector<int>& A) {
        int len = A.size();
        vector<int> res;
        if( len == 0 )
            return res;
        res = vector<int>(len, 1);
        for( int i=0;i<len;i++ )
            for( int j=0;j<len;j++ )
            {
                if( j == i )
                    continue;
                res[i] *= A[j];
            }
        
        return res;
    }
};

方法二

可以将B[i]=A[0]*A[1]*...*A[i-1]*A[i+1]*...*A[n-1]看作是两部分的乘积，前半部分记作first，后半部分记作second，B[i] = first[i] * second[i]

• first[i] = A[0]*A[1]*...*A[i-1]，可以利用公式first[i] = first[i-1] * A[i-1]求出first（从前往后算）
• second[i] = A[i+1]*...*A[n-1]，可以利用公式second[i] = second[i+1] * A[i+1]求出second（算的时候从后往前算）
• B[i] = first[i] * second[i]

可以看出方法的复杂度为O(n)，比方法一复杂度低

代码

class Solution {
public:
    vector<int> multiply(const vector<int>& A) {
        int len = A.size();
        vector<int> res;
        if( len == 0 )
            return res;
        res = vector<int>(len, 1);
        
        // 求前半部分的积
        vector<int> first(len, 1);
        for( int i=1;i<len;i++ )
            first[i] = first[i-1] * A[i-1];
        
        // 求后半部分的积
        vector<int> second(len, 1);
        for( int i=len-2;i>=0;i-- )
            second[i] = second[i+1] * A[i+1];
        
        // 求B[i]
        for( int i=0;i<len;i++ )
            res[i] = first[i] * second[i];
        
        return res;
    }
};