差分(前缀和的逆运算)

#include <iostream>

using namespace std;

const int N = 100;


// *一维差分
// 
// 差分可以看作前缀和的逆运算
// 有一个前缀和数组 S[n], 则可以得到一个数组 a[n] = S[n] - S[n-1]
// a[n] 数组称为 S[n] 数组的差分
// 
// 
// *主要应用*(重点掌握)
// S[n] 的 [l, r] 区间都加上 c，对 a[n] 数组的影响是：a[l] += c, a[r + 1] -= c

int S[N], a[N];
void insert_one(int l, int r, int c) // (重点掌握)
{
	a[l] += c;

	if (r + 1 < N) // 防止越界访问
	{
		a[r + 1] -= c;
	}
}



// *二维差分*
// 
// 有一个前缀和数组 S[n][n], 则可以得到一个数组 a[n] = S[n][n] - S[n][n-1] - S[n-1][n] + S[n-1][n-1]
// 
// *主要应用*(重点掌握)
// 给 S[n][n] 中以(x1, y1)为左上角，(x2, y2)为右下角的子矩阵中的所有元素加上 c，对 a[n][n] 数组的影响是：a[x1][y1] += c
//																								   a[x2 + 1][y2 + 1] += c
//																								   a[x2 + 1][y1] -= c 
//																								   a[x1][y2 + 1] -= c
//

int SS[N][N], aa[N][N];
void insert_two(int x1, int y1, int x2, int y2, int c) // (重点掌握)
{
	aa[x1][y1] += c;
	aa[x2 + 1][y2 + 1] += c;
	aa[x1][y2 + 1] -= c;
	aa[x2 + 1][y1] -= c;
}

int main()
{
	// 一维差分
	for (int i = 1; i < N; i++)
	{
		scanf("%d", &S[i]); // 给定前缀和数组
	}

	for (int i = 1; i < N; i++)
	{
		insert_one(i, i, S[i]);// 假设原 S[n] 数组都为 0，则原 a[n] 数组也全为 0，此操作相当于求出 S[n] 对应的差分 a[n]
	}



	// 二维差分
	for (int i = 1; i < N; i++)
	{
		for (int j = 1; j < N; j++)
		{
			scanf("%d", &SS[i][j]);
		}
	}

	for (int i = 1; i < N; i++)
	{
		for (int j = 1; j < N; j++)
		{
			insert_two(i, j, i, j, SS[i][j]); // 假设原 S[n][n] 数组都为 0，则原 a[n][n] 数组也全为 0，此操作相当于求出 S[n][n] 对应的差分 a[n][n]
		}
	}
	return 0;
}