差分矩阵python3

差分矩阵

输入一个 nn 行 mm 列的整数矩阵，再输入 qq 个操作，每个操作包含五个整数 x1,y1,x2,y2,cx1,y1,x2,y2,c，其中 (x1,y1)(x1,y1) 和 (x2,y2)(x2,y2) 表示一个子矩阵的左上角坐标和右下角坐标。

每个操作都要将选中的子矩阵中的每个元素的值加上 cc。

请你将进行完所有操作后的矩阵输出。

输入格式
第一行包含整数 n,m,qn,m,q。

接下来 nn 行，每行包含 mm 个整数，表示整数矩阵。

接下来 qq 行，每行包含 55 个整数 x1,y1,x2,y2,cx1,y1,x2,y2,c，表示一个操作。

输出格式
共 nn 行，每行 mm 个整数，表示所有操作进行完毕后的最终矩阵。

数据范围
1≤n,m≤10001≤n,m≤1000,
1≤q≤1000001≤q≤100000,
1≤x1≤x2≤n1≤x1≤x2≤n,
1≤y1≤y2≤m1≤y1≤y2≤m,
−1000≤c≤1000−1000≤c≤1000,
−1000≤矩阵内元素的值≤1000−1000≤矩阵内元素的值≤1000
输入样例：
3 4 3
1 2 2 1
3 2 2 1
1 1 1 1
1 1 2 2 1
1 3 2 3 2
3 1 3 4 1
输出样例：
2 3 4 1
4 3 4 1
2 2 2 2

#对原矩阵aij
#构造差分矩阵bij：假设A全0则B也全0，又在A的ij位置+aij，B相应insert再求前缀和

#A是B的前缀和，矩阵左上角

#将A的子矩阵(x1,y1)~(x2,y2)位置部分+C，
# 即把B(x1,y1)+C,左上角加了C，其前缀和A的整个右下角都加了C
# 再B(x1,y2+1)-C，B(x2+1,y1)-C
# 再B(x1+1,y1+1)+C
#即：左上右下+C，左下右上-C，但要注意是否+1
def insert(B,x1,y1,x2,y2,c):
    B[x1][y1]+=c
    B[x2+1][y2+1]+=c
    B[x1][y2+1]-=c
    B[x2+1][y1]-=c


if __name__=="__main__":
    n,m,q=map(int,input().split())
    A=[[0]*(m+10) for i in range(n+10)]#A为(n+1)*(m+1)的全0矩阵，这里多写一点0，尺寸大一点，可以保证不出下标超出界限错
    B=[[0]*(m+10) for i in range(n+10)]
    for i in range(n):#i=0,...,n 存输入的矩阵A 因为不能像意味的直接加[0]，所以只能用赋值方法写
        A[i+1][1:]=list(map(int,input().split()))#注意这里下标的表达

    for i in range(1,n+1):#把全0的A变成A，则对B以下操作
        for j in range(1,m+1):
            insert(B,i,j,i,j,A[i][j])#Aij的插入，即在全0的A的子矩阵[i,j]-[i,j]位置+C

    for i in range(q):#A的子矩阵+C，对b操作如下
        x1,y1,x2,y2,c=map(int,input().split())
        insert(B,x1,y1,x2,y2,c)

    for i in range(1,n+1):#对B求前缀和
        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=" ")#不能直接B输出，因为一开始B就多定义了很多0
        print()#使得end=换行