前言
前缀和,你一串长度为n的数列a1,a2,a3…an,再给出m个询问,每次询问给出L,R两个数,要求给出区间[L,R]里的数的和。
差分,给你一串长度为n的数列a1,a2,a3…an,要求对a[L]~a[R]进行m次操作:
操作一:将a[L]~a[R]内的元素都加上P
操作二:将a[L]~a[R]内的元素都减去P
最后再给出一个询问求a[L]-a[R]内的元素之和?
一、前缀和
一维前缀和
#include<bits/stdc++.h>
using namespace std;
int a[100005];
int sum[100005];
int main()
{
int n,m;
cin>>n>>m;
for(int i=1;i<=n;i++)
{
cin>>a[i];
}
for(int i=1;i<=n;i++)
{
sum[i]=sum[i-1]+a[i];
}
while(m--)
{
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=1010;
int a[N][N];
int sum[N][N];
int main()
{
int n,m,q;
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++)
{
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[x1-1][y2]-sum[x2][y1-1]+sum[x1-1][y1-1]<<endl;
}
return 0;
}
二、差分
一维差分
#include<bits/stdc++.h>
using namespace std;
//给定a[1],a[2],....a[n];
//构造差分数组b[N];
//a[i]=b[1]+b[2]+...+b[i];
//核心操作:将a[L~R]全部加上C,等价于
//b[L]+=C,b[R+1]-=C;
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];
insert(i,i,a[i]);//构造差分数组
}
while(m--)
{
int l,r,c;
cin>>l>>r>>c;
insert(l,r,c);
}
for(int i=1;i<=n;i++)
{
a[i]=a[i-1]+b[i];
printf("%d ",a[i]);
}
puts("");
return 0;
}
二维差分
#include<bits/stdc++.h>
using namespace std;
//给定原矩阵a[i,j],构造差分矩阵b[i,j];
//使得a[i,j]是b[i,j]的二维前缀和
//核心操作:给(x1,y1)为左上角,(x2,y2)为 右下角,所有的数,加上C;
//s[x1,y1]+=c;
//s[x1,y2+1]-=c;
//s[x2+1,y1]-=c;
//s[x2+1,y2+1]+=c;
const int N=1010;
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()
{
int n,m,q;
cin>>n>>m>>q;
for(int i=1;i<=n;i++)
{
for(int j=1;j<=m;j++)
{
cin>>a[i][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++)
{
a[i][j]=a[i-1][j]+a[i][j-1]-a[i-1][j-1]+b[i][j];
printf("%d ",a[i][j]);
}
puts("");
}
return 0;
}