一维前缀和和二维前缀和

一、 一维前缀和

作用：快速求出数组中一段数的和

树状数组和线段树也可以快速求出数组中一段数的和，但是前缀和时间复杂度很低，查询效率高，但是前缀和只能查询不能修改，其他两个能边查询边修改

具体原理（数列前n项和公式）：
$S_r=a_1+a_2+...+a_l+...+a_r$
$S_{l-1}=a_1+a_2+...+a_{l-1}$
$S_r-S_{l-1}=a_l+...+a_r$

注意：前缀和的数组下标从1开始

使用：给定一个数组，求出数组a[l]+…+a[r]的和

#include <iostream>      
#include <algorithm>      
#include <cstring>      
#include <vector>

using namespace std;

const int N = 100010;

int n, m;
// 原数组 a
int a[N]
// 前缀和数组 s
int s[N];


int main()
{
	cin >> n >> m;
	// 读取数组元素
	for (int i = 1; i <= n; i++)
	{
		cin >> a[i];
		// 计算前缀和
		s[i] = s[i - 1] + a[i];
	}

	// 处理每个询问
	while (m-- > 0)
	{
		int l, r;
		cin >> l >> r;
		// 打印数组a 从a[l]+ a[l+1] + ... + a[r]的和
		cout << s[r] - s[l - 1] << endl;
	}

	return 0;
}

二、二维前缀和

原数组和二维前缀和数组的关系

二维前缀和数组[2][2] = 17，它是原二维数组左上角(1,1)到右下角(2,2)所有数的和

通过原数组计算二维前缀和数组

由容斥原理计算二维前缀和数组s[2][3]
等于原数组红色矩阵部分所有数的和
等于原数组中蓝色部分+紫色部分-黄色部分+灰色部分
黄色部分相当于加了2次所以要减去黄色部分

紫色部分=前缀和数组s[1][3]的值
蓝色部分=前缀和数组s[2][2]的值
黄色部分=前缀和数组s[1][2]的值
灰色部分=原数组a[2][3]的值

所以$$s[2][3]=[1][3]+s[2][2]-s[1][2]+a[1][1]$$

所以二维前缀和公式：
$$s[x][y]=s[x-1][y]+s[x][y-1]-s[x-1][y-1]+a[x][y]$$

注意：数组下标需要从1开始，且$s[0][0]=0$, $a[0][0]=0$

公式记忆

三、子矩阵和

利用前缀和数组计算原数组子矩阵的和

也是利用容斥原理
求出原数组以(2,2)为左上角坐标，(3,3)为右下角坐标的子矩阵中所有数的和

黄色部分面积=蓝色部分面积s[3][3]-橙色部分面积s[3][1]-粉色部分面积s[1][3]+灰色部分面积s[1][1] （灰色部分被减了两次所以要加回来）
$=s[3][3]-s[3][1]-s[1][3]+s[1][1]$

所以子矩阵和公式$$[x1y1,x2y2]=s[x2][y2]-s[x2][y1-1]-s[x1-1][y2]+s[x1-1][y1-1]$$

[x1y1,x2y2]是以x1,y1为左上角坐标，以x2,y2为右下角坐标的矩形面积

#include <iostream>      
using namespace std;
const int N = 1010;
// n行m列 q个询问
int n, m, q;
// 原数组a  前缀和数组s 
int a[N][N], s[N][N];

int main()
{
	ios::sync_with_stdio(false);
	cin.tie(0);
	cin >> n >> m >> q;
	// 读入每个元素
	for (int i = 1; i <= n; i++)
	{
		for (int j = 1; j <= m; j++)
		{
			cin >> a[i][j];
			// 求出前缀和数组s
			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;
		cin >> x1 >> y1 >> x2 >> y2;
		// 计算原矩阵中以x1,y1为左上角 x2,y2为右上角的的矩阵的和
		cout << s[x2][y2] - s[x1 - 1][y2] - s[x2][y1 - 1] + s[x1 - 1][y1 - 1] << endl;
	}
	return 0;
}