UVA-10827 环面上的最大子矩阵和

uva-10827 环面上的最大子矩阵和

题目描述:

把二维矩阵的第一行和最后一行,第一列和最后一列连起来,得到一个环面,求此矩阵上的最大子矩阵(即其中所有元素之和最大)
题目链接:https://vjudge.net/problem/UVA-10827

参考:https://blog.csdn.net/tengfei461807914/article/details/54808927
上述参考中使用c++可以通过,使用python好像会超时

def find(matrix):
    ext_matrix = matrix + matrix
    ext_matrix = [v*2 for v in ext_matrix]
    # print(ext_matrix)
    row = len(ext_matrix)//2
    res = float("-inf")
    for i in range(row):
        for j in range(row):
            dp = [0]*row*2
            for m in range(i,row+i):
                tmp = 0
                for n in range(j,row+j):
                    dp[n] += ext_matrix[m][n]
                    tmp += dp[n]
                    res = max(res,tmp)
    return res

case_num = int(input())
for i in range(case_num):
    matrix = []
    N = int(input())
    for j in range(N):
        tmp = list(map(int,input().strip().split()))
        matrix.append(tmp)
    print(find(matrix))

posted @ 2019-07-23 00:41  静_渊  阅读(183)  评论(0编辑  收藏  举报