求子矩阵价值最大的左上角坐标（python）

# 求子矩阵最大价值（所有值相加），返回左上角坐标
# 暴力做法  会超时
# 输入： 3 4 2      输出：1 2
#        1 2 3 1
#       -1 9 0 2
#        2 0 1 1

n,m,c=map(int,input().split())
a=[[]for i in range(n)]
for i in range(n):  # 读取矩阵
    a[i]=list(map(int,input().split()))
    
val=-float("inf") # 初始化为最小值
best_x,best_y=1,1  # 初始化坐标
for i in range(n-c+1):  # 子矩阵行范围
    for j in range(m-c+1):  # 子矩阵列范围
        total_val=0
        for u in range(c):  # 开始扫描子矩阵
            for v in range(c):
                total_val+=a[i+u][j+v]  # 计算子矩阵价值
        if total_val>val:
            val=total_val
            best_x,best_y=i+1,j+1
print(best_x,best_y)