专题三:前缀和与差分
目录
前言
本专题将介绍前缀和与差分技巧,主要讲解其原理以及算法模板,再枚举一些例题和真题,加深大家对于该算法的理解,帮助大家在赛场上游刃有余地运用该算法!
前缀和
定义
前缀和是指某序列的前n项和,可以把它理解为数学上的数列的前n项和(等差数列),而差分可以看成前缀和的逆运算。合理的使用前缀和与差分,可以将某些复杂的问题简单化,降低时间复杂度。
分类
- 一维前缀和
- 二维前缀和
一维前缀和
如何理解?
假设有一个原数组a,如何求出前缀和数组s
a[1],a[2],a[3].....a[n]
s[i] = a[i] + a[i-1]...a[2] + a[1]
a[3] + a[4]...a[14] + a[15] = s[15] - s[3-1]
s[l,r] = s[r] - s[l-1]
模板
S[i] = a[1] + a[2] + ... a[i]
a[l] + ... + a[r] = S[r] - S[l - 1]
例题
代码
本题是对模板的直接运用。
#include<bits/stdc++.h>
using namespace std;
const int N=100010;
int n,m;
int a[N],s[N];
int main()
{
cin>>n>>m;
for(int i=1;i<=n;i++)
cin>>a[i];
for(int i=1;i<=n;i++)
s[i]=s[i-1]+a[i];
while(m--)
{
int l,r;
cin>>l>>r;
cout<<s[r]-s[l-1]<<endl;
}
}
二维前缀和
如何理解?
给大家看一张我认为比较好理解的图。
输入一个n行m列的整数矩阵,再输入q个询问,每个询问包含四个整数x1, y1, x2, y2,表示一个子矩阵的左上角坐标和右下角坐标。对于每个询问输出子矩阵中所有数的和。
同一维前缀和一样,我们先来定义一个二维数组s[][] , s[i][j] 表示二维数组中,左上角(1, 1)到右下角(i, j)所包围的矩阵元素的和。接下来推导二维前缀和的公式。
从图中我们很容易看出,整个外围蓝色矩形面积s[i][j] = 绿色面积s[i - 1][j] + 紫色面积s[i][j - 1] - 重复加的红色的面积s[i - 1][j - 1] + 小方块的面积a[i][j];
因此得出二维前缀和预处理公式
s[i][j] = s[i - 1][j] + s[i][j - 1 ] + a[i] [j] - s[i - 1][j - 1]
接下来回归问题去求以(x1,y1)
为左上角和以(x2,y2)
为右下角的矩阵的元素的和。
因此二维前缀和的结论为:
以(x1, y1)
为左上角,(x2, y2)
为右下角的子矩阵的和为:s[x2, y2] - s[x1 - 1, y2] - s[x2, y1 - 1] + s[x1 - 1, y1 - 1]
例题
代码
一个数组写法
#include <iostream>
using namespace std;
const int N = 1010;
int n, m, q;
int s[N][N];
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", &s[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];
while (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;
}
两个数组写法
#include<bits/stdc++.h>
using namespace std;
const int N=1010;
int a[N][N];
int s[N][N];
int n,m,q;
int main()
{
cin>>n>>m>>q;
for(int i=1;i<=n;i++)
for(int j=1;j<=m;j++)
{
cin>>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;
cin>>x1>>y1>>x2>>y2;
cout<<s[x2][y2]-s[x1-1][y2]-s[x2][y1-1]+s[x1-1][y1-1]<<endl;
}
}
差分
一维差分
怎么理解?
用图来辅助理解
模板
直接给出步骤
一维差分:本质为a数组是b数组的前缀和数组
1. 给区间内的每个数加上c: B[l]+=c;B[r+1]-=c;
2. 差分为前缀和的逆运算,构造差分数组(b[i]=a[i]-a[i-1])
3. 求前缀和:b[i]+=b[i-1]输出b数组或者a[i]=a[i-1]+b[i]输出a数组
例题
代码
#include<bits/stdc++.h>
using namespace std;
const int N=100010;
int n,m;
int a[N],b[N];
void insert(int l,int r,int c)
{
b[l]+=c;
b[r+1]-=c;
}
int main()
{
cin>>n>>m;
for(int i=1;i<=n;i++) cin>>a[i];
for(int i=1;i<=n;i++)
{
b[i]=a[i]-a[i-1];//构造差分数组
// insert(i,i,a[i]);
}
while(m--)
{
int l,r,c;
cin>>l>>r>>c;
b[l]+=c,b[r+1]-=c;
}
for(int i=1;i<=n;i++) b[i]+=b[i-1];//求出前缀和
for(int i=1;i<=n;i++) cout<<b[i]<<" ";
return 0;
}
二维差分
如何理解?
难点在于构造!一维差分我们很容易就可以通过b[i]=a[i]-a[i-1]
来构造 差分数组,但是二分就不行了,如果还是按照这种思维会很复杂。那么正确的方式是什么?看一张图
例题
代码
#include<bits/stdc++.h>
using namespace std;
const int N=1010;
int n,m,q;
int a[N][N],b[N][N];
//差分模板
void insert(int x1,int y1,int x2,int y2,int c)
{
b[x1][y1]+=c;
b[x1][y2+1]-=c;
b[x2+1][y1]-=c;
b[x2+1][y2+1]+=c;
}
int main()
{
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++)
insert(i,j,i,j,a[i][j]);//构造二维差分数组
while(q--)
{
int x1,y1,x2,y2,c;
cin>>x1>>y1>>x2>>y2>>c;
insert(x1,y1,x2,y2,c);
}
for(int i=1;i<=n;i++)
for(int j=1;j<=m;j++)
{
b[i][j]+=b[i-1][j]+b[i][j-1]-b[i-1][j-1];
}
for(int i=1;i<=n;i++)
{
for(int j=1;j<=m;j++)
{
cout<<b[i][j]<<" ";
}
cout<<endl;
}
}
截断数组
代码
#include<bits/stdc++.h>
using namespace std;
const int N=100010;
int s[N];
int n;
int main()
{
cin>>n;
for(int i=1;i<=n;i++)
{
int x;
cin>>x;
s[i]=s[i-1]+x;
}
if(s[n]%3) puts("0");
else{
long long cnt=0,res=0;
for(int j=2;j<n;j++)
{
if(s[j-1]==s[n]/3) cnt++;
if(s[j]==s[n]/3*2) res+=cnt;
}
cout<<res<<endl;
}
return 0;
}
思路
因为要把数组均分三份,总和一定得是3的倍数
把整个分割方案模拟成一个集合,将分割的子集相加即为答案。
枚举固定住第二个截断的点,分割方案就是第一个点可以取的位置,且由于分割的数组分空:第二个分割点必须要保证前面有两个以上的元素才行。
真题:统计子矩阵
代码
#include <iostream>
using namespace std;
const int N = 510;
int n, m, k, a[N][N], s[N][N], res;
int main()
{
cin >> n >> m >> k;
// 输入矩阵A
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];
// 枚举所有子矩阵
for (int i = 1; i <= n; i ++ )
for (int j = 1; j <= m; j ++ )
for (int x = i; x <= n; x ++ )
for (int y = j; y <= m; y ++ )
{
int t = s[x][y] - s[i - 1][y] - s[x][j - 1] + s[i - 1][j - 1];
if (t <= k) res ++ ;
}
cout << res << endl;
return 0;
}
总结
本专题主要讲解了差分与前缀和算法,大家需要重点去掌握一维、二维的前缀和与差分的算法模板以及如何推导出公式的理解,只要掌握了数学推导才能真正的去理解并且运用到其他题目上,预祝大家在蓝桥杯赛场取得好成绩。