【AcWing】前缀和（前缀和模板）

Lg_970

已于 2022-03-21 17:38:56 修改

阅读量551

点赞数 3

文章标签：算法 c++

于 2022-03-13 13:59:56 首次发布

本文链接：https://blog.csdn.net/L_NanN/article/details/123457144

版权

前缀和模板

前提：只能处理静态数组，只能查询不能修改。

预处理：通过S0,S1,S2,…,Sn分别存储前缀和：
S0=0、S1=a1、S2=a1+a2、… 、Sn=a1+a2+…+an。
则当我们多次查询某段[l, r] 区间内元素总和 时，可以通过Sr-Sl-1 求得al+…+ar。
优点：避免了每次对某段遍历求和，降低了时间复杂度。
局限性：只能处理静态数组，无法修改数组。
什么时候使用前缀和：大量地重复（询问）查询数组内某段元素之和的时候。

题目1:模板题

AcWing 前缀和

#include <bits/stdc++.h>
using namespace std;

int a[100010],s[100010]; //定义成全局数组，初值必定为0
int main(){
    int n,m;
    cin >> n >> m;
    for(int i=1;i<=n;i++){
        cin >> a[i];
        s[i]=s[i-1]+a[i];
    }
    while(m--){
        int l,r;
        cin >> l >> r;
        cout << s[r]-s[l-1] << endl;
    }
    return 0;
}

题目2：前缀和统计出现次数

AcWing 出现次数

#include <bits/stdc++.h>
using namespace std;

int a[1010],s[1010];//a数组记录：str1下标i开始长度为m的子串是否==str2

int main(){
    int n,m,q;
    string str1,str2;
    cin >> n >> m >> q >> str1 >> str2;
    for(int i=0;i+m<=n&&i<n;i++){ //i+m<=n保证str1剩下长度不足m时退出
        if(str1.substr(i,m)==str2) a[i]=1; //说明i这个位置是一个str2出现的开始位置
    }
    for(int i=0;i<n;i++) s[i+1]=s[i]+a[i]; //前缀和数组s，s[i]表示在str1的i位置上，（往前+当前）已经出现了s[i]个str2
    while(q--){
        int l,r;
        scanf("%d %d",&l,&r);
        if(r-l+1<m) cout << 0 << endl; //str1子串长度小于m，不可能出现
        else cout << s[r-m+1]-s[l-1] << endl; //r-m+1刚好在右边界前最后可能出现str2的位置上，s[r-m+1]即可表示这个位置已经出现了多少个str2，不能用s[r]因为r的位置上可能往后还会出现str2，但这个str2不在l~r范围内
    }
    return 0;
}

字符串find函数用法

同时学习了string find函数的用法：
find函数如果没有找到，返回string::npos。
find函数如果找到，返回子串dest在str中的下标。
然后在循环体内将pos++，使下次find从新pos开始。
如此重复find，直到pos==string::npos

	int pos=0;
	string str,dest; 
	cin >> str >> dest;
    while((pos=str.find(dest,pos))!=string::npos){ //在str中查找dest
		cout << pos << endl; //如果找到，输出pos
		pos++;
	}

在这里插入图片描述

题目3：前缀和矩阵

AcWing 子矩阵的和

思想：

快速求一个矩阵的任意子矩阵的和：
先预处理，构造前缀和矩阵，该矩阵上的Sij表示原矩阵对应位aij左上方（包括aij）的子矩阵和。
构造前缀和矩阵，采用容斥原理： $S_{i,j}=S_{i-1,j}+S_{i,j-1}-S_{i-1,j-1}+a_{i,j}$ 。
通过前缀和矩阵求子矩阵的和，采用容斥原理： $S_{x_2,y_2}-S_{x_1-1,y_2}-S_{x_2,y_1-1}+S_{x_1-1,y_1-1}$ 。

#include <bits/stdc++.h>
using namespace std;
const int N = 1010;
int a[N][N],s[N][N];
int main(){
	int n,m,q;
	scanf("%d%d%d",&n,&m,&q);
	for(int i=1;i<=n;i++){ //注意：i，j要从1开始，因为要存在s[0][0]=0
		for(int j=1;j<=m;j++){
			scanf("%d",&a[i][j]);
			s[i][j]=s[i-1][j]+s[i][j-1]-s[i-1][j-1]+a[i][j]; //构造前缀和矩阵
		}
	}
	while(q--){ //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;
}