前缀和
前缀和:将数组a[i]的和求一遍。时间复杂度由O(n)*O(l-r)变成O(n)*O(1)。
前缀和s[i]=a[1]+a[2]+a[3]+...+a[i](数组求和)
题目:
输入一个长度为 n𝑛 的整数序列。
接下来再输入 m𝑚 个询问,每个询问输入一对 l,r𝑙,𝑟。
对于每个询问,输出原序列中从第 l𝑙 个数到第 r𝑟 个数的和。
输入格式
第一行包含两个整数 n𝑛 和 m𝑚。
第二行包含 n𝑛 个整数,表示整数数列。
接下来 m𝑚 行,每行包含两个整数 l𝑙 和 r𝑟,表示一个询问的区间范围。
输出格式
共 m𝑚 行,每行输出一个询问的结果。
数据范围
1≤l≤r≤n1≤𝑙≤𝑟≤𝑛,
1≤n,m≤1000001≤𝑛,𝑚≤100000,
−1000≤数列中元素的值≤1000−1000≤数列中元素的值≤1000
输入样例:
5 3
2 1 3 6 4
1 2
1 3
2 4
输出样例:
3
6
10
//前缀和解法:
#include<iostream>
using namespace std;
const int N=1e5+10;
int sum[N]={0},a[N];
int main()
{
int m,n;
cin >> m >> n;
for(int i=1;i<=m;i++) cin >> a[i];
for(int i=1;i<=m;i++) sum[i]=sum[i-1]+a[i];
while(n--)
{
int l,r;
cin >> l >> r;
cout << sum[r]-sum[l-1] << endl;
}
return 0;
}
//下面是暴力解法:
/*
#include<bits/stdc++.h>
using namespace std;
const int N=1e5+10;
int n,m;
int a[N],sum[N];
int i1=0;
int add(int a[],int i,int j)
{
int s=0;
for(;i<=j;i++)
s+=a[i];
return s;
}
int main()
{
cin >> m >> n;
for(int i=0;i<m;i++) cin >> a[i];
while(n--)
{
int k,l;
cin >> k >> l;
sum[i1++]=add(a,k-1,l-1);
}
int j=0;
while(sum[j]!=0)
{
cout << sum[j] << endl;
j++;
}
return 0;
}
*/
子矩阵的和
也是利用前缀和求解,不过二维处理起来较麻烦
下面附上我的理解:
公式:s[l,r]=s[l-1,r]+s[l,r-1]-s[l-1,r-1]+a[l,r]
(不再推导,思路大致如上)
注意‼ :此时的s[i,j]表示的全为矩阵的和。那么问题来了最原始的矩阵怎么表示,也就
是说如果前面的s[i,j-1]的矩阵和该怎么求???
hh,这里我想了一会,原因就是因为在第一行中的s[1,j]中的原始前缀和都为矩阵,所以
后面的s[i,j]可以依次向前递归求解。
题目:
输入一个 n𝑛 行 m𝑚 列的整数矩阵,再输入 q𝑞 个询问,每个询问包含四个整数 x1,y1,x2,y2𝑥1,𝑦1,𝑥2,𝑦2,表示一个子矩阵的左上角坐标和右下角坐标。
对于每个询问输出子矩阵中所有数的和。
输入格式
第一行包含三个整数 n,m,q𝑛,𝑚,𝑞。
接下来 n𝑛 行,每行包含 m𝑚 个整数,表示整数矩阵。
接下来 q𝑞 行,每行包含四个整数 x1,y1,x2,y2𝑥1,𝑦1,𝑥2,𝑦2,表示一组询问。
输出格式
共 q𝑞 行,每行输出一个询问的结果。
数据范围
1≤n,m≤10001≤𝑛,𝑚≤1000,
1≤q≤2000001≤𝑞≤200000,
1≤x1≤x2≤n1≤𝑥1≤𝑥2≤𝑛,
1≤y1≤y2≤m1≤𝑦1≤𝑦2≤𝑚,
−1000≤矩阵内元素的值≤1000−1000≤矩阵内元素的值≤1000
输入样例:
3 4 3
1 7 2 4
3 6 2 8
2 1 2 3
1 1 2 2
2 1 3 4
1 3 3 4
输出样例:
17
27
21
代码如下:
#include<iostream>
using namespace std;
int a[1050][1050];
//不需要开long long 最大值为数组全为1000,则sum最大值为1e9。不会爆int
//二维求和数组长度与数组长度一直就行;
int sum[1050][10050]={0};
//sum长度开的太大会超内存--Memory Limit Exceed
//int sum[20050][20050]={0};
int main()
{
int m,n,q;
cin >> m >> n >> q;
//注意从i,j=1开始
for(int i=1;i<=m;i++)
for(int j=1;j<=n;j++)
cin >> a[i][j];
//注意从i,j=1开始
for(int i=1;i<=m;i++)
for(int j=1;j<=n;j++)
sum[i][j]=sum[i-1][j]+sum[i][j-1]-sum[i-1][j-1]+a[i][j];
while(q--)
{
int x1,y1,x2,y2;
cin >> x1 >> y1 >> x2 >> y2;
//矩阵公式
cout << sum[x2][y2]-sum[x2][y1-1]-sum[x1-1][y2]+sum[x1-1][y1-1] << endl;
}
return 0;
}
差分
差分这个理解起来可能比较抽象,是构造出来一个数组b,令a中的每一个值都是数组b的前缀和
b[1]=a[1],b[2]=a[2]-a[1],b[3]=a[3]-a[2]...b[i]=a[i]-a[i-1]。
所以a[1]=a[1],a[2]=b[1]+b[2],a[3]=b[1]+b[2]+b[3]...a[i]=b[1]+b[2]+b[3]+...+b[i]
那么如果b[3]加上了c,a[3]后(包括a[3])的每一个值都会加上c;此时再让b[10]减去c,那么a[10]后(包括a[10])的每一个值都减去c,这样就能满足a[3]~a[10]之间(包括边界)的数都加上了c。
题目:
输入一个长度为 n𝑛 的整数序列。
接下来输入 m𝑚 个操作,每个操作包含三个整数 l,r,c𝑙,𝑟,𝑐,表示将序列中 [l,r][𝑙,𝑟] 之间的每个数加上 c𝑐。
请你输出进行完所有操作后的序列。
输入格式
第一行包含两个整数 n𝑛 和 m𝑚。
第二行包含 n𝑛 个整数,表示整数序列。
接下来 m𝑚 行,每行包含三个整数 l,r,c𝑙,𝑟,𝑐,表示一个操作。
输出格式
共一行,包含 n𝑛 个整数,表示最终序列。
数据范围
1≤n,m≤1000001≤𝑛,𝑚≤100000,
1≤l≤r≤n1≤𝑙≤𝑟≤𝑛,
−1000≤c≤1000−1000≤𝑐≤1000,
−1000≤整数序列中元素的值≤1000−1000≤整数序列中元素的值≤1000
输入样例:
6 3
1 2 2 1 2 1
1 3 1
3 5 1
1 6 1
输出样例:
3 4 5 3 4 2
代码:
#include<iostream>
using namespace std;
const int N=1e5+10;
int a[N],b[N];
/*
将a[i]看成是数组b[i]的前缀和,所以当计算b[i]的前缀和时,(l,n)上的数a[i]会自动加上c,
(r+1,n)上的a[i]会减去c。所以总体而言(l,r)上的数据会加上c
*/
void insert(int l,int r,int c)
{
b[l]+=c;
//这里的r+1就是将数组a成为数组b前缀和的关键
b[r+1]-=c;
}
int main()
{
int n,m;
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++) scanf("%d",&a[i]);
/*这一步是求b[i](由于a[i]是b[i]的前缀和,所以根据前缀和可以计算出b[i]),
insert函数是为了方便对(l,r)上a[i]加上c
*/
for(int i=1;i<=n;i++) insert(i,i,a[i]);
while(m--)
{
int l,r,c;
scanf("%d%d%d",&l,&r,&c);
insert(l,r,c);
}
for(int i=1;i<=n;i++)
{
b[i]+=b[i-1];//应为:a[i]=b[i]+b[i-1];
printf("%d ",b[i]);
}
return 0;
}
差分矩阵
思路大致相同,不过就是把差分,前缀和,矩阵综合起来而已。
题目:
输入一个 n𝑛 行 m𝑚 列的整数矩阵,再输入 q𝑞 个操作,每个操作包含五个整数 x1,y1,x2,y2,c𝑥1,𝑦1,𝑥2,𝑦2,𝑐,其中 (x1,y1)(𝑥1,𝑦1) 和 (x2,y2)(𝑥2,𝑦2) 表示一个子矩阵的左上角坐标和右下角坐标。
每个操作都要将选中的子矩阵中的每个元素的值加上 c𝑐。
请你将进行完所有操作后的矩阵输出。
输入格式
第一行包含整数 n,m,q𝑛,𝑚,𝑞。
接下来 n𝑛 行,每行包含 m𝑚 个整数,表示整数矩阵。
接下来 q𝑞 行,每行包含 55 个整数 x1,y1,x2,y2,c𝑥1,𝑦1,𝑥2,𝑦2,𝑐,表示一个操作。
输出格式
共 n𝑛 行,每行 m𝑚 个整数,表示所有操作进行完毕后的最终矩阵。
数据范围
1≤n,m≤10001≤𝑛,𝑚≤1000,
1≤q≤1000001≤𝑞≤100000,
1≤x1≤x2≤n1≤𝑥1≤𝑥2≤𝑛,
1≤y1≤y2≤m1≤𝑦1≤𝑦2≤𝑚,
−1000≤c≤1000−1000≤𝑐≤1000,
−1000≤矩阵内元素的值≤1000−1000≤矩阵内元素的值≤1000
输入样例:
3 4 3
1 2 2 1
3 2 2 1
1 1 1 1
1 1 2 2 1
1 3 2 3 2
3 1 3 4 1
输出样例:
2 3 4 1
4 3 4 1
2 2 2 2
代码:
#include<iostream>
using namespace std;
int m,n,q;
const int N=1010;
int a[N][N],b[N][N];
//板子,将数组a变成b的前缀和
void insert(int x1,int y1,int x2,int y2,int c)
{
b[x1][y1]+=c;
b[x2+1][y1]-=c;
b[x1][y2+1]-=c;
b[x2+1][y2+1]+=c;
}
int main()
{
scanf("%d%d%d",&m,&n,&q);
for(int i=1;i<=m;i++)
for(int j=1;j<=n;j++)
scanf("%d",&a[i][j]);
for(int i=1;i<=m;i++)
for(int j=1;j<=n;j++)
insert(i,j,i,j,a[i][j]);
//对某一子矩阵进行操作
while(q--)
{
int x1,y1,x2,y2,c;
scanf("%d%d%d%d%d",&x1,&y1,&x2,&y2,&c);
insert(x1,y1,x2,y2,c);
}
for(int i=1;i<=m;i++)
for(int j=1;j<=n;j++)
{
//前缀和公式,对b求前缀和
a[i][j]=b[i][j]+a[i-1][j]+a[i][j-1]-a[i-1][j-1];
printf("%d ",a[i][j]);
if(j==n) printf("\n");
}
return 0;
}
板子:
前缀和:
sum[i]=sum[i-1]+a[i];
子矩阵的和:
求矩阵的和:sum[i][j]=sum[i-1][j]+sum[i][j-1]-sum[i-1][j-1]+a[i][j]
求某一子矩阵的和:sum[x2][y2]-sum[x1-1][y2]-sum[x2][y1-1]+sum[x1-1][y1-1]
差分:
void insert(int l,int r,int c)
{
b[l]+=c;
b[r+1]-=c;
}
差分矩阵:
void insert(int x1,int y1,int x2,int y2,int c)
{
b[x1][y1]+=c;
b[x2+1][y1]-=c;
b[x1][y2+1]-=c;
b[x2+1][y2+1]+=c;
}
tips
一、数据输入输出量较大时,建议使用scanf,printf 输入输出,效率更高。
二、板子需要理解着去记,不然你只会这一题...
三、针对本节模块去刷题
四、不理解的一定要去自己动手推导,不然你永远也不会...