AcWing 798. 差分矩阵(python)
原题链接
简单
作者:
等待_02
,
2024-01-20 12:19:53
,
所有人可见
,
阅读 26
if __name__ == "__main__":
n, m, q = map(int, input().split())
# 提前添加一行,这样后续不需要判断边界
matrix = [[0 for i in range(m+2)]]
b = [[0] * 1010 for i in range(1010)]
# 先用简单思路构建一个矩阵,按行构建差分矩阵
for i in range(1, n + 1):
s = [0]
s.extend(list(map(int, input().split())))
s.append(0)
matrix.append(s)
# 获取完矩阵后下边界也要设置好
matrix.append([0 for i in range(m+2)])
# 构建差分矩阵
for i in range(1, n + 1):
for j in range(1, m + 1):
c = matrix[i][j]
b[i][j] += c; b[i][j + 1] -= c; b[i + 1][j] -= c; b[i + 1][j + 1] += c
for i in range(q):
x1, y1, x2, y2, c = map(int, input().split())
b[x1][y1] += c; b[x1][y2 + 1] -= c; b[x2 + 1][y1] -= c; b[x2 + 1][y2 + 1] += c
for i in range(1, n + 1):
for j in range(1, m + 1):
b[i][j] = b[i][j - 1] + b[i - 1][j] - b[i - 1][j - 1] + b[i][j]
print(b[i][j], end=' ')
print()