程序员面试题（二）

继续写面试题系列。

输出一个矩阵中的不同行不同列的最大和。比如

7	53	183	439	√863
497	√383	563	79	973
287	63	√343	169	583
627	343	773	√959	943
√767	473	103	669	303

比如上面那个矩阵中打对勾的数值的和就是符合要求的。

这个问题我当时没有答出来，主要不知道如何去遍历，其实用三层迭代就可以实现，如果过几天有时间，我会为此写一篇博文。今天就用一个取巧的办法。

那天是我收到我在北京的最后一个拒绝，在准备回来的路上想到的。提前把所有的情况遍历出来不就好了？

就好比上述那个数组，其实就是建立一个数组[0, 1, 2, 3, 4]，然后每行从这里面分别取值即可，也就是这个数组元素的所有排列。

手动写一个排列，也就是我上面所说的三层迭代，是个挺麻烦的事情。这里就借助itertools中的permutations模块啦。

all_per = permutations(xrange(columns), columns)

这样就解决了。把这其中的每一种情况遍历即可。

def max_sum(matrix):
    rows = len(matrix)
    columns = len(matrix[0])
    all_per = permutations(xrange(columns), columns)
    sum_list = []
    
    for each_turn in all_per:
        for idx in xrange(columns):
            sum_each = sum([matrix[row][each_turn[idx]] for row in xrange(rows)])
            sum_list.append(sum_each)
    return max(sum_list)

就这样一个问题，我竟然也没有答出来 ==。

源代码：Github