一维和二维前缀和

————出自南昌理工学院ACM集训队

一维前缀和

基本思想

前缀和是什么呢？前缀就是一个数组的某项下标之前(包括此项元素)的所有数组元素的和。
简单点来说就是有n个数，求n个数包括第n个数的所有数的和。
S[n]=a[1]+a[2]+a[3]+a[4]+…+a[n].
前缀和最基本的用法就是解决求某个区间所有数的和，经过前缀和的预处理，可以大大降低查询的时间复杂度。

模板

//前缀和的预处理
for(int i=0;i<=n;i++){//b[i]存i之前和i所有的数之和
    b[i]=b[i-1]+a[i];
}
int l,r;//区间的范围
cin>>l>>r;
cout<<b[r]-b[l-1];//求l到r的所有的数的和，因为是求[l,r]，包括了l，所以只需减掉l-1的所有的数之和即可。

例题

简单的原题
题目
输入一个长度为n的整数序列。
接下来再输入m个询问，每个询问输入一对l, r。
对于每个询问，输出原序列中从第l个数到第r个数的和。
输入格式
第一行包含两个整数n和m。
第二行包含n个整数，表示整数数列。
接下来m行，每行包含两个整数l和r，表示一个询问的区间范围。
输出格式
共m行，每行输出一个询问的结果。
数据范围
1≤l≤r≤n ,
1≤n,m≤100000,
−1000≤数列中元素的值≤1000
输入样例：

5 3
2 1 3 6 4
1 2
1 3
2 4

输出样例

3
6
10

思路
就是求n到m区间的所有数的和，直接用前缀和求解即可。

代码：

#include <iostream>
using namespace std;
typedef long long ll;
const int N = 1e5+5;
int main()
{
    int n,m,a[N];
    scanf("%d %d", &n, &m);
    for (int i = 1; i <= n; i++) {
        scanf("%d", &a[i]);//输入数据
    }
    a[0]=0;
    for (int i = 1; i <= n; i++) {
        a[i] = a[i] + a[i - 1];//前i个和第i个总和
    }
    while (m--) {
        int x, y;
        scanf("%d %d", &x, &y);
        cout << a[y] - a[x-1] << endl;//区间[x,y]的和
    }
    return 0;
}

了解了一维前缀和，就来两个例题练练手吧：最短距离，数圈圈。

二维前缀和

基本思想

二维前缀和其实和一维的思想差不多.

基本公式是：a[i][j]=a[i-1][j]+a[i][j-1]-a[i-1][j-1]+a[i][j] ，
方便理解我们直接上图来解释
以下是数组a[6][5]所存的数：

例如我们要求a[3][4]前缀和，本身肯定要相加，我们还得再加上a[2][4]的前缀和,再加上a[3][3]的前缀和，然后我们会发现a[2][3]的这个部分加了两遍，所以我们还要减去一次a[2][3]的值，所以我们得出a[i][j]=a[i-1][j]+a[i][j-1]-a[i-1][j-1]+a[i][j] 这个公式。

模板

//前缀和的预处理
for(int i=1;i<=n;i++){
   for(int j=1;j<=m;j++){
      a[i][j]=a[i-1][j]+a[i][j-1]-a[i-1][j-1]+a[i][j]; //s[i][j]用来存储a的前缀和，用a[i][j]来存也是一样的
   }
}

例题

（原题）
题目
输入一个n行m列的整数矩阵，再输入q个询问，每个询问包含四个整数x1, y1, x2, y2，表示一个子矩阵的左上角坐标和右下角坐标。
对于每个询问输出子矩阵中所有数的和。
输入格式
第一行包含三个整数n，m，q。
接下来n行，每行包含m个整数，表示整数矩阵。
接下来q行，每行包含四个整数x1, y1, x2, y2，表示一组询问。
输出格式
共q行，每行输出一个询问的结果。
数据范围
1≤n,m≤1000,
1≤q≤200000,
1≤x1≤x2≤n,
1≤y1≤y2≤m,
−1000≤矩阵内元素的值≤1000
输入样例

3 4 3
1 7 2 4
3 6 2 8
2 1 2 3
1 1 2 2
2 1 3 4
1 3 3 4

输出样例

17
27
21

思路

就是给定一个区间，求这个区间二维矩阵的所有数之和，象征性的来说就是求红色区域的面积，直接用总面积减去黄色和绿色的面积，再加上重复减的面积，也就是前缀和。
代码

#include <iostream>
#include <algorithm>
#include <cstring>
using namespace std;
typedef long long ll;
const int N=1010;
int a[N][N],s[N][N];
int n,m,q;
int main(){
    cin>>n>>m>>q;
    for(int i=1;i<=n;i++){//输入
    	for(int j=1;j<=m;j++){
    		cin>>a[i][j];
		}
	}
	for(int i=1;i<=n;i++){//前缀和的预处理
		for(int j=1;j<=m;j++){
			a[i][j]=a[i-1][j]+a[i][j-1]-a[i-1][j-1]+a[i][j]; //模板
		}
	}
	for(int i=0;i<q;i++){
		int x1,y1,x2,y2;
		cin>>x1>>y1>>x2>>y2;
		cout<<a[x2][y2]-a[x1-1][y2]-a[x2][y1-1]+a[x1-1][y1-1]<<endl;//总面积-左边部分-上面部分+重合的部分
	}
    return 0;
}

emm，没了