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))