蓝桥杯2022年第十三届省赛真题-统计子矩阵

不知道为什么指针越界？

def Pre_sum(sum,z):
    for g in range(1, m + 1):
        for h in range(1, n + 1):
            sum[g][h] = sum[g - 1][h] + sum[g][h - 1] - sum[g - 1][h - 1] + z[g][h]
    return sum

# 从左上角到右下角
def Submatrix(sum,x1,y1,x2,y2):
    return sum[x2][y2]-sum[x1-1][y2]-sum[x2][y1-1]+sum[x1-1][y1-1]


m,n,k = map(int, input().split())
c=0
z=[[0]*(n+1) for q in range(m+1)]
sum=[[0]*(n+1) for q in range(m+1)]
for p in range(1,m+1):
    z[p]=[0]+list(map(int, input().split()))

sum=Pre_sum(sum,z)

# 左上角
for x1 in range(1,m+1):
    for y1 in range(1,n+1):
        # 右下角
        for x2 in (x1,m+1):
            for y2 in (y1,n+1):
                if Submatrix(sum,x1,y1,x2,y2)<=k:
                    c+=1
print(c)