ET-Campus算法竞赛_1_ 前缀和与差分

1. 前缀和

题目实例一：一维前缀和

P1049 - 【模板】前缀和 - ETOJ (eriktse.com)

给定一个长度为n的数组a，有q次询问，每次询问数组a在区间 [ l , r ] 的和。

#include <bits/stdc++.h>
using namespace std;

using ll = long long;
const int N = 1e5 + 9;
ll a[N], prefix[N];

int main()
{
	ios::sync_with_stdio(0), cin.tie(0), cout.tie(0);

	int n; cin >> n;
	for (int i = 1; i <= n; ++i) cin >> a[i];
	for (int i = 1; i <= n; ++i) prefix[i] = prefix[i - 1] + a[i];

	int q; cin >> q;
	while (q--) {
		int l, r; cin >> l >> r;
		cout << prefix[r] - prefix[l - 1] << "\n";
	}
}

特性

任意子段和可由两端的前缀和相减得到

具有良好的修改性质

题目实例二：二维前缀和

P1060 - 【模板】二维前缀和 - ETOJ (eriktse.com)

给定一个n行m列的整数矩阵。

有q次询问，每次询问格式为：x1​,y1​,x2​,y2​，表示一个子矩阵的左上角和右下角的坐标。

对于每个询问，请回答子矩阵的所有数之和

#include <bits/stdc++.h>
using namespace std;
using ll = long long;

const int N = 1e3 + 9;
ll a[N][N], prefix[N][N];

int main()
{
	ios::sync_with_stdio(0), cin.tie(0), cout.tie(0);

	int m, n, q; cin >> m >> n >> q;
	for (int i = 1; i <= m; ++i)
		for (int j = 1; j <= n; ++j)
			cin >> a[i][j];
	for (int i = 1; i <= m; ++i)
		for (int j = 1; j <= n; ++j)
			prefix[i][j] = prefix[i - 1][j] + prefix[i][j - 1] - prefix[i - 1][j - 1] + a[i][j];

	while (q--) {
		int x1, y1, x2, y2; cin >> x1 >> y1 >> x2 >> y2;
		cout << prefix[x2][y2] - prefix[x2][y1 - 1] - prefix[x1 - 1][y2] + prefix[x1 - 1][y1 - 1] << '\n';
	}
}

2.差分

题目实例一：一维差分

P1050 - 【模板】差分 - ETOJ (eriktse.com)

给定一个长度为n的数组a

有m次修改，每次修改让数组a在区间 [ l , r ]增加x

在修改完成后有q次询问，每次询问数组在区间 [ l , r ] 的和

#include <bits/stdc++.h>
using namespace std;

using ll = long long;
const int N = 1e5 + 9;
ll a[N], diff[N], prefix[N];

int main()
{
	ios::sync_with_stdio(0), cin.tie(0), cout.tie(0);
	int n; cin >> n;
	for (int i = 1; i <= n; ++i) cin >> a[i];
	for (int i = 1; i <= n; ++i) diff[i] = a[i] - a[i - 1];

	int m; cin >> m;
	while (m--) {
		int l, r; cin >> l >> r;
		ll x; cin >> x;
		diff[l] += x; diff[r + 1] -= x;
	}

	for (int i = 1; i <= n; ++i) a[i] = a[i - 1] + diff[i];
	for (int i = 1; i <= n; ++i) prefix[i] = prefix[i - 1] + a[i];

	int q; cin >> q;
	while (q--) {
		int l, r; cin >> l >> r;
		cout << prefix[r] - prefix[l - 1] << '\n';
	}
}

特性

累加可还原为原数组（前缀和的逆过程）

可进行后缀区间修改（静态，即修改完成再询问）

具有良好的修改性质

题目实例二：二维差分

P1061 - 【模板】二维差分 - ETOJ (eriktse.com)

给定一个n行m列的整数矩阵。
有q个操作，每个操作格式为：x1​,y1​,x2​,y2​,c，其中(x1,y1)、(x2,y2)分别表示一个子矩阵的左上角和右下角的坐标，每个操作将对应的子矩阵的每个元素加上c。
请输出进行完所有操作后的矩阵。

#include <bits/stdc++.h>
using namespace std;
using ll = long long;

const int N = 1e3 + 9;
ll a[N][N], diff[N][N];

int main()
{
	ios::sync_with_stdio(0), cin.tie(0), cout.tie(0);

	int m, n, q; cin >> m >> n >> q;
	for (int i = 1; i <= m; ++i)
		for (int j = 1; j <= n; ++j)
			cin >> a[i][j];
	for (int i = 1; i <= m; ++i)
		for (int j = 1; j <= n; ++j)
			diff[i][j] = a[i][j] - a[i - 1][j] - a[i][j - 1] + a[i - 1][j - 1];

	while (q--) {
		int x1, y1, x2, y2, c; cin >> x1 >> y1 >> x2 >> y2 >> c;
		diff[x1][y1] += c;
		diff[x2 + 1][y1] -= c;
		diff[x1][y2 + 1] -= c;
		diff[x2 + 1][y2 + 1] += c;
	}

	for (int i = 1; i <= m; ++i) {
		for (int j = 1; j <= n; ++j) {
			a[i][j] = a[i - 1][j] + a[i][j - 1] - a[i - 1][j - 1] + diff[i][j];
			cout << a[i][j] << ' ';
		}
		cout << '\n';
	}
}