ACWING 前缀和 差分 差分矩阵 python注释版

一维前缀和

'''
输入一个长度为 n的整数序列。
接下来再输入 m个询问，每个询问输入一对 l , r
对于每个询问，输出原序列中从第 l 个数到第 r 个数的和。

输入格式
第一行包含两个整数 n 和 m。
第二行包含 n个整数，表示整数数列。
接下来 m行，每行包含两个整数 l和 r，表示一个询问的区间范围。

输出格式
共 m行，每行输出一个询问的结果。

数据范围
1 ≤ l ≤ r ≤ n
1 ≤ n , m ≤ 100000 ,
− 1000 ≤ ≤ 数列中元素的值 ≤ 1000

输入样例：
5 3
2 1 3 6 4
1 2
1 3
2 4
输出样例：
3
6
10
'''
# 一维前缀和
# 输入数列长度为n m次询问
n,m = map(int,input().split())

# 构造输入数组
a=[0]*(n+1)
a[1:n+1] = list(map(int,input().split()))

# 构造输出数组
for i in range(1,n+1):
    a[i]+=a[i-1]

# 计算l,r之间的和

for i in range(m):
    l, r = map(int, input().split())
    print(a[r]-a[l-1])

二维前缀和

# 二维前缀和
# 输入n行m列 q次询问
n,m,q = map(int,input().split())

# 构造初始数组
a=[[0]*(m+1) for _ in range(n+1)]
for i in range(1,n+1):
    a[i][1:]= list(map(int,input().split()))

# 构造前缀和数组
for i in range(1,n+1):
    for j in range(1,m+1):
        a[i][j] += a[i-1][j]+a[i][j-1]-a[i-1][j-1]

# 计算前缀和
for _ in range(q):
    x1,y1,x2,y2 = map(int,input().split())
    ans = a[x2][y2]-a[x1-1][y2]-a[x2][y1-1]+a[x1-1][y1-1]
    print(ans)

一维差分

'''
输入一个长度为 n 的整数序列。
接下来输入 m 个操作，每个操作包含三个整数 l,r,c，表示将序列中 [l,r] 之间的每个数加上 c。
请你输出进行完所有操作后的序列。

输入格式
第一行包含两个整数 n 和 m。
第二行包含 n 个整数，表示整数序列。
接下来 m 行，每行包含三个整数 l，r，c，表示一个操作。

输出格式
共一行，包含 n 个整数，表示最终序列。

数据范围
1≤n,m≤100000,
1≤l≤r≤n,
−1000≤c≤1000,
−1000≤整数序列中元素的值≤1000
输入样例：
6 3
1 2 2 1 2 1
1 3 1
3 5 1
1 6 1
输出样例：
3 4 5 3 4 2
'''

'''
思路：
原数组A' 差分数组B
先假定原数组A全为0，则差分数组B也全为0
将原数组A'用插入操作插入到全为0的原数组A中(初始化)
然后根据题目要求的将序列中 [l,r] 之间的每个数加上 c(即插入操作)
再求出数组B的前缀和则得到完成所有操作的原数组A''(前缀和与差分类似于逆运算)
'''
# 数列长度为n m次询问
n, m = map(int, input().split())

# 初始化原数组A和差分数组
a = [0] * (n + 2)
b = [0] * (n + 2)

# 初始化原数组A‘
a[1:n+1] = list(map(int,input().split()))

# 定义插入函数
def insert(l,r,c):
    b[l]+=c
    b[r+1]-=c

# 将A’使用插入操作到A中
for i in range(1,n+1):
    insert(i,i,a[i])

# 将序列A’中 [l,r] 之间的每个数加上 c(在差分数组B中进行插入操作来完成这一要求）
for i in range(m):
    l,r,c = map(int,input().split())
    insert(l,r,c)

# 计算B的前缀和得到最终要得到的序列A''
for i in range(1,n+1):
    b[i]+=b[i-1]
    print(b[i],end=" ")

二维差分

'''
输入一个 n 行 m 列的整数矩阵，再输入 q 个操作，每个操作包含五个整数 x1, y1, x2, y2, c,
其中 (x1, y1) 和 (x2, y2) 表示一个子矩阵的左上角坐标和右下角坐标。每个操作都要将选中的子矩阵中的每个元素的值加上 c。
请你将进行完所有操作后的矩阵输出。

输入格式
第一行包含整数n,m,q。
接下来n行，每行包含m个整数，表示整数矩阵。
接下来q行，每行包含5个整数x1, y1, x2, y2, c，表示一个操作。

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

数据范围
1≤n,m≤1000,
1≤q≤100000,
1≤x1≤x2≤n,
1≤y1≤y2≤m,
−1000≤c≤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 18
2 2 2 2
'''

# 数组长度为n行m列 q次询问
n, m, q = map(int, input().split())

# 创建原数组和差分数组
a = [[0] * (m + 2) for _ in range(n + 2)]
b = [[0] * (m + 2) for _ in range(n + 2)]


# 定义insert函数
def insert(x1, y1, x2, y2, c):
    b[x1][y1] += c
    b[x2 + 1][y1] -= c
    b[x1][y2 + 1] -= c
    b[x2 + 1][y2 + 1] += c


# 初始化原数组
for i in range(1, n + 1):
    a[i][1:m + 1] = list(map(int, input().split()))

# 初始化差分数组
for i in range(1, n + 1):
    for j in range(1, m + 1):
        insert(i, j, i, j, a[i][j])

# q次插入操作
for i in range(q):
    x1, y1, x2, y2, c = map(int, input().split())
    insert(x1, y1, x2, y2, c)

# 计算最终数组
for i in range(1, n + 1):
    for j in range(1, m + 1):
        b[i][j] += b[i - 1][j] + b[i][j - 1] - b[i - 1][j - 1]
        print(b[i][j], end=" ")
    print()