前缀与差分（用C/C++实现）

一、前缀

前缀和：有一数组为a(1), a(2), ……, a(n)，则前缀和S(i) = a(1) + a(2) + …… + a(i).

1. 求法：使用for循环累加，S(0)定义为0.
2. 作用：用于计算数组中任意某段的和，如[l, r]，可使用S(r) – S(l - 1)计算.

/*输入长度为n的整数序列，输入m个询问每个询问输入一对il, r
对于每行询问，输出原序列中[l, r]的和*/
#include<iostream>

using namespace std;

const int N = 1e6 + 10;

int n, m;
int a[N], s[N];

int main()
{
	//ios::sync_with_stdio(false);//让标准输入输出无需同步，能够提高cin的读取速度，但无法使用scanf了 
	scanf("%d %d", &n, &m);
	for(int i = 1; i <= n; i++) scanf("%d", &a[i]);
	
	for(int i = 1; i <= n; i++) s[i] = s[i - 1] + a[i];
	
	while(m --)
	{
		int l, r;
		scanf("%d %d", &l, &r);
		printf("%d\n", s[r] - s[l - 1]);
	}
	return 0;
}

二维前缀和：求子矩阵的和（两方向的前缀）

（容斥原理思想）以上图中小矩阵为例，小方阵中元素的和可拆分为四部分：大方阵减去两个矩形在加上重复减去的区域。

二、差分（前缀和的逆运算）

1. 一维差分

原数组为a(1), a(2), ……, a(n).

构造新数组b(1), b(2), ……, b(n)，使得a(i) = b(1) + …… + b(i).

构造方法： b = a(1).

                        b(2) = a(2) – a(1).

                        b(3) = a(3) – a(2).

                               ……

                        b(n) = a(n) – a(n - 1).

*假定原本的数组全为0，之后做了n次插入操作，如元素一为[1, 1] + a(1)，以此类推。故差分只需要对b中两个元素进行操作。

作用：只要有b数组，就能快速求得a数组。若要在a数组的区间[l, r]中使每一个元素加c，则只需使得b(l)+c，且使b(r+1)-c

/*输入一个长度为n的序列，输入m个操作
每个操作包含l, r, c，表示将序列[l, r]之间的每个数加上c
输出完成操作的数组*/
#include<iostream>

using namespace std;

const int N = 100010;

int n, m;
int a[N], b[N];

void insert(int l, int r, int c)
{
	b[l] += c;
	b[r + 1] -= c;
}

int main()
{
	scanf("%d %d", &n, &m);
	for(int i = 1; i <= n; i ++) scanf("%d", &a[i]);
	
	for(int i = 1; i <= n; i ++) insert(i, i, a[i]);
	
	while(m --)
	{
		int l, r, c;
		scanf("%d %d %d", &l, &r, &c);
		insert(l, r, c);
		
	}
	for(int i = 1; i <= n; i ++) b[i] += b[i - 1];//结果即为b数组的前缀和 
	
	for(int i = 1; i <= n; i ++) printf("%d ", b[i]);
	
	return 0;
} 

  2. 二维差分

#include<iostream>

using namespace std;

const int N = 10010;

int n, m, q;
int a[N][N], b[N][N];
void insert(int x1, int y1, int x2, int y2, int c)
{
	b[x1][y1] += c;
	b[x2 + 1][y1] -= c;
	b[x1][y2 + 1] -= c;
	b[x2 + 1][y2 + 1] += c;
}

int main()
{
	scanf("%d %d %d", &n, &m, &q);
	
	for(int i = 1; i <= n; i ++)
	{
		for(int j = 1; j <= m; j ++)
		{
			scanf("%d", &a[i][j]);
		}
	}
	
	for(int i = 1; i <= n; i ++)
	{
		for(int j = 1; j <= m; j ++)
		{
			insert(i, j, i, j, a[i][j]);
		}
	}
	while(q --)
	{
		int x1, y1, x2, y2, c;
		cin >> x1 >> y1 >> x2 >> y2 >> c;
		insert(x1, y1, x2, y2, c);
	}
	for(int i = 1; i <= n; i ++)
	{
		for(int j = 1; j <= m; j ++)
		{
			b[i][j] +=  b[i - 1][j] + b[i][j - 1] - b[i - 1][j -1];
		}
	}
	for(int i = 1; i <= n; i ++)
	{
		for(int j = 1; j <= m; j ++) printf("%d ", b[i][j]);
		puts("");
	}
	
	return 0;
}