前缀和与差分

目录

一，前缀和

 二，差分

一维数组差分

 二维数组差分

---

一，前缀和

因为一维前缀和比较简单，就没写了，直接看二维前缀和吧，二维前缀和用了容斥原理

代码如图：

#include<iostream>                            //二维前缀和模板
#include<algorithm>

using namespace std;

const int N = 1010;

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

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++)
		{
			s[i][j] = s[i - 1][j] + s[i][j - 1] - s[i - 1][j - 1] + a[i][j];
		}
	}
	while (q--)
	{
		int x1, y1, x2, y2;
		scanf("%d%d%d%d", &x1, &y1, &x2, &y2);
		printf("%d\n", s[x2][y2] - s[x1 - 1][y2] - s[x2][y1 - 1] + s[x1 - 1][y1 - 1]);
	}
	return 0;
}

 二，差分

差分就是前缀和的逆运算，用于求解对一个数组的一段区间上加上某个数

一维数组差分

代码如图：

#include<iostream>         //一维差分   用于求解在一个数组中的一段区间上加上某个数的问题
#include<algorithm>

using namespace std;

const int N = 1e5 + 10;

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

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];         //对差分数组求一遍前缀和得到原数组
	for (int i = 1; i <= n; i++)
		printf("%d ", b[i]);
	return 0;
}

 二维数组差分

如图：

#include<iostream>                  //二维数组差分  
#include<algorithm>
#include<cstdio>
using namespace std;

const int N = 1e4 + 10;

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

void insert(int x1, int y1, int x2, int y2, int c)
{
	b[x1][y1] += c;                //搞清楚哪里加哪里减
	b[x1][y2 + 1] -= c;
	b[x2 + 1][y1] -= 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;
		scanf("%d%d%d%d%d", &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]);
		}
		printf("\n");
	}
	return 0;
}