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

Reference Answer

思路分析

没什么难度，直接上代码。

# -*- coding:utf-8 -*-
class Solution:
    def multiply(self, A):
        # write code here
        res = A[:]
        temp_prod = 1
        for index,count in enumerate(A):
            res[index] = self.product(A[:index] + A[index+1:])
        return res
    def product(self, A):
        temp_prod = 1
        for index,count in enumerate(A):
            temp_prod *= count
        return temp_prod

其中上式的 product 函数可以直接用内嵌的 reduce 替代：

# -*- coding:utf-8 -*-
class Solution:
    def multiply(self, A):
        # write code here
        if not A:
            return []
        res = A[:]
        for index, count in enumerate(A):
            res[index] = reduce(lambda x,y:x*y, A[:index]+ A[index+1:])
        return res

再提供一种解题方案

思路

观察下公式，你会发现，B[i]公式中没有A[i]项，也就是说如果可以使用除法，就可以用公式B[i]=A[0]A[1]…*A[n-1]/A[i]来计算B[i]，但是题目要求不能使用，因此我们只能另想办法。

现在要求不能使用除法，只能用其他方法。一个直观的解法是用连乘n-1个数字得到B[i]。显然这个方法需要O(n*n)的时间复杂度。

好在还有更高效的算法。可以把B[i]=A[0]A[1]…*A[i-1]A[i+1]…*A[n-1]。看成A[0]A[1]…A[i-1]和A[i+1]…A[n-2]*A[n-1]两部分的乘积。

即通过A[i]项将B[i]分为两部分的乘积。效果如下图所示：

不妨设定C[i]=A[0]A[1]…A[i-1]，D[i]=A[i+1]…*A[n-2]*A[n-1]。C[i]可以用自上而下的顺序计算出来，即C[i]=C[i-1]*A[i-1]。类似的，D[i]可以用自下而上的顺序计算出来，即D[i]=D[i+1]*A[i+1]。

如果还是不明白，没有关系，直接看下代码，细细体会下就懂了。

第一个for循环用来计算上图1范围的数，第二个for循环用来计算上图2范围的数。

C++ version:

class Solution {
public:
    vector<int> multiply(const vector<int>& A) {
        int length = A.size();
        vector<int> B(length);
        if(length <= 0){
            return B;
        }
        B[0] = 1;
        for(int i = 1; i < length; i++){
            B[i] = B[i-1]*A[i-1];
        }
        int temp = 1;
        for(int i = length-2; i >= 0; i--){
            temp *= A[i+1];
            B[i] *= temp;
        }
        return B;
    }
};

Note:

• 上述这种构建过程中利用到了两种思路，一种是直接过滤掉 index 对应值进行计算，一种是计算分成 index 前与 index 后两者乘积计算。
• 其中要善于利用 python 自带 reduce 函数，如本题： reduce(lambda x,y:x*y, A[:index]+ A[index+1:])，很方便。