【算法学习】前缀和

最新推荐文章于 2024-09-14 21:53:48 发布

小菜陌噫嗷的成长之路

最新推荐文章于 2024-09-14 21:53:48 发布

阅读量82

点赞数

文章标签：学习算法

本文链接：https://blog.csdn.net/mm1840817359/article/details/131249801

版权

一、基本思想

【一维】

长度为n的数组：a1，a2，……an

前缀和数组：Si=a1+a2+……+ai s[0]=0（统一，比较容易处理边界）

1、如何求Si？

S[i]=S[i-1]+a[i];

2、Si的作用

能快速地求出原数组中一段数组地和，如求区间 [l，r] 这段数组的和，为S[r]-S[l-1]；

【二维】

长度为n的数组：a[1][1]，a[2][1]，……a[n][n]；

前缀和数组：s[i][j]=a[1][1] + a[2][1] + ……+ a[n][n]；

1、如何求s[i][j]?

i(1~n) j(1~m)

s[i][j]=s[i-1][j] + s[i][j-1] - s[i-1][j-1] + a[i][j]；

2、s[i][j]的作用？

用来快速求出子矩阵的和；

3、求子矩阵的公式

黑色阴影小矩形各元素的和=s[x2][y2] - s[x2][y1-1] - s[x1-1][y2] + s[x1-1][y1-1]；

二、代码模板

//一维前缀和
S[i] = a[1] + a[2] + ... a[i]
a[l] + ... + a[r] = S[r] - S[l - 1]

//二维前缀和
S[i, j] = 第i行j列格子左上部分所有元素的和
以(x1, y1)为左上角，(x2, y2)为右下角的子矩阵的和为：
S[x2, y2] - S[x1 - 1, y2] - S[x2, y1 - 1] + S[x1 - 1, y1 - 1]

三、经典例题

【例一】前缀和

题目描述：

代码：

#include<bits/stdc++.h>
using namespace std;
const int N=100010;

int n,m;
int a[N];
int s[N];
int res=0;

int main()
{
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++)
	{
		scanf("%d",&a[i]);
	}
	s[0]=0;
	for(int i=1;i<=n;i++)
	{
		s[i]=s[i-1]+a[i];  //前缀和初始化
	}
	while(m--)         //依次处理
	{
		int l,r;
		scanf("%d%d",&l,&r);
		res=s[r]-s[l-1];   //区间和的计算
		printf("%d\n",res);
	}
	return 0;
}

【例二】子矩阵的和

题目描述：

代码：

#include<bits/stdc++.h>
using namespace std;
const int N=1010;

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

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]);
			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);
		res=s[x2][y2]-s[x2][y1-1]-s[x1-1][y2]+s[x1-1][y1-1];
		printf("%d\n",res);
	}
	return 0;
}