前缀和+求子矩阵和

前缀和数组: si=a1+a2+a3+…+ai
1.前缀和求法 s[i]=s[i-1]+a[i];
2.作用:求[l,r]一段区间的和 s[r]-s[l-1]
3.i最好从1开始,这样s[1]=s[0]+a[1];

题目链接:https://www.acwing.com/problem/content/797/
代码:

#include<iostream>
using namespace std;

const int N=1e5+5;
int n,m;
int a[N], s[N];
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++) 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;
} 

模板变型:求子矩阵和
题目链接:https://www.acwing.com/problem/content/798/

#include<iostream>
using namespace std;
const int maxn=1e3+5;
int a[maxn][maxn], s[maxn][maxn];

int main(){
	int n,m,q;
	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++)
			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;
		//注意这个公式容易出错,写的时候画画思考一下过程
		cout<<s[x2][y2]-s[x1-1][y2]-s[x2][y1-1]+s[x1-1][y1-1]<<endl;
	} 
	return 0;
}
©️2020 CSDN 皮肤主题: 游动-白 设计师:上身试试 返回首页