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