前缀和基础(学习记录)

前缀和和差分之前就讲过了,有一段时间没用过了,之前学习的时候也是只做了几个板子题,现在有点遗忘了,再做几个题复习一下吧,,,

以几个板子题  1657 前缀和Ⅰ  1658 前缀和Ⅱ  1659 前缀和Ⅲ  1660 前缀和Ⅳ 为例复习一下前缀和,应用写到下一篇里吧~差分要在后面了QAQ


前缀和:指序列中前n项的和,可以类比数学中数列的前n项和,只不过数学中数列的前n项和在这里是一维前缀和,我们要研究的除了一维,还有二维。前缀和可以视为一种预处理操作,在某些题中可以简化问题。

一维前缀和

若有序列 a[0] , a[1] , a[2] , a[3] ...... a[n] ,根据前面定义可知,前缀和数组 dif为:

           dif[0] = a[0] ; dif[1] = a[0] + a[1] ; dif[2] = a[0] + a[1] + a[2] ......

           dif[n] = a[0] + a[1] + a[2] ...... + a[n]

时间复杂度:预处理(即得到前缀和数组)O(n),查询(即求某一段前缀和)O(1);

用具体数字举例(是师哥上课用的PPT里的):

对于a数组,其下标为index,sum为前缀和数组。 

那么我们使用前缀和数组可以做什么呢,对于这样一个简单的问题:对于长度为n的整数序列,有m个询问,对于输入的每组l和r,输出第l个数到第r个数的和。若不用前缀和预处理,我们可能会选用for循环直接遍历求和,但这样是O(n*m)的时间复杂度,有TLE的风险,我们若使用前缀和处理,可以大大减少时间复杂度。

对于一维前缀和,我们常用以下两公式,sum即前缀和数组,ans即询问区间求和结果:

 (不知道会不会有人有这样的疑问:为什么不是减去sum[l]而是减去sum[l-1]?我们求第三个数到第五个数的和,当然是用前五个数的和减去前两个数的和啦)

一维前缀和板子题 1657 前缀和Ⅰ

AC代码(最简单的板子题): 

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <stack>
#include <vector>
#include <map>
#include <queue>
#include <cstring>
#include <cmath>
#include <set>
#include <iterator>

using namespace std;

typedef long long ll;
#define ios ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
#define INF 0x3f3f3f3f
//判断是否需要开ll!!!
//判断是否需要初始化!!!
const int mod=1e9+7;
const int N=100005;
int n,q,p;
int a[N];
ll dif[N];//注意判断数据范围,前缀和数组需要开ll! 

int main()
{
//	freopen("test.in","r",stdin);
    cin>>n>>q;//对于n个数的序列,有q次询问
    for(int i=1;i<=n;i++)
    {
    	cin>>a[i];
    	dif[i]=dif[i-1]+a[i];//从1开始输入有个好处:似乎不用考虑dif[1]的值。。。 
	}
	while(q--)
	{
		cin>>p;//输出前p个数的和
		cout<<dif[p]<<'\n';
	}
	return 0;
}

求一维区间前缀和的板子题  1658 前缀和Ⅱ

用到了前面的ans求答案。。。太怪了这个题居然卡cin/cout。。。QAQ

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <stack>
#include <vector>
#include <map>
#include <queue>
#include <cstring>
#include <cmath>
#include <set>
#include <iterator>

using namespace std;

typedef long long ll;
#define ios ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
#define INF 0x3f3f3f3f
//判断是否需要开ll!!!
//判断是否需要初始化!!!
const int mod=1e9+7;
const int N=100005;
int n,q,l,r;
int a[N];
ll dif[N];
ll ans;

int main()
{
//	freopen("test.in","r",stdin);
    scanf("%d%d",&n,&q);
    for(int i=1;i<=n;i++)
    {
    	scanf("%d",&a[i]);
    	dif[i]=dif[i-1]+a[i];
	}
	while(q--)
	{
		scanf("%d%d",&l,&r);
		ans=dif[r]-dif[l-1];//注意减去的是什么!
		printf("%lld\n",ans);
	}
	return 0;
}

二维前缀和

一维理解起来较容易,那么我们来看二维的前缀和。怎样求下图中橙色部分的和?

 那么类比一维前缀和数组构造,可构造二维前缀和数组:

通俗点说就是:sum[i][j] = sum[i-1][j] + sum[i][j-1] + a[i][j] - sum[i-1][j-1] 

上图的二维前缀和矩阵则为:

 在计算二维数组区间和之前,为了防止计算重复,我们需要了解一个计数原理:容斥原理。

容斥原理:在计数时,必须注意没有重复,没有遗漏。为了使重叠部分不被重复计算,人们研究出一种新的计数方法,这种方法的基本思想是:先不考虑重叠的情况,把包含于某内容中的所有对象的数目先计算出来,然后再把计数时重复计算的数目排斥出去,使得计算的结果既无遗漏又无重复,这种计数的方法称为容斥原理。(百度百科yyds!)

如下图,求橙色区域和我们需要通过其他几种颜色的区域和来转化,具体公式:

                 ans = sum[x2][y2] - sum[x2][y1-1] - sum[x1-1][y2] + sum[x1-1][x2-1]

 

求二维数组板子题  1659 前缀和Ⅲ

AC代码:

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <stack>
#include <vector>
#include <map>
#include <queue>
#include <cstring>
#include <cmath>
#include <set>
#include <iterator>

using namespace std;

typedef long long ll;
#define ios ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
#define INF 0x3f3f3f3f
//判断是否需要开ll!!!
//判断是否需要初始化!!!
const int mod=1e9+7;
const int N=1005;
int n,m,q,x,y;
int a[N][N];
ll dif[N][N];//前缀和二维数组

int main()
{
//	freopen("test.in","r",stdin);
    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]);
    		dif[i][j]=dif[i][j-1]+dif[i-1][j]+a[i][j]-dif[i-1][j-1];
		}
	}
	while(q--)
	{
		scanf("%d%d",&x,&y);
		printf("%lld\n",dif[x][y]);
	}
	return 0;
}

这个二维数组前缀和用到了ans,,,1660 前缀和Ⅳ

AC代码:

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <stack>
#include <vector>
#include <map>
#include <queue>
#include <cstring>
#include <set>
#include <iterator>

using namespace std;

typedef long long ll;
#define ios ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
#define INF 0x3f3f3f3f
//判断是否需要开ll!!!
//判断是否需要初始化!!!
const int mod=1e9+7;
const int N=1005;
int n,m,q;
int x1,y1,x2,y2;
int a[N][N];
ll dif[N][N];
ll ans;//记得开ll!

int main()
{
//	freopen("test.in","r",stdin);
    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]);
    		dif[i][j]=dif[i-1][j]+dif[i][j-1]+a[i][j]-dif[i-1][j-1];
		}
	}
	while(q--)
	{
		scanf("%d%d%d%d",&x1,&y1,&x2,&y2);
		ans=dif[x2][y2]-dif[x2][y1-1]-dif[x1-1][y2]+dif[x1-1][y1-1];
		printf("%lld\n",ans);
	}
	return 0;
}

基础知识点常看常新,多复习多复习~

如有错误请赐教!

  Orz

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值