今天搞了搞前缀和和差分的操作,感觉还是很棒的,这个唯一的不足就是不可以修改
前缀和:
一维前缀和
int a[N];
int tr[N];
void insert(int i,int t)
{
tr[i]=tr[i-1]+t;
}
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++)
insert(i,a[i]);
while(m--)
{
int l,r;
cin>>l>>r;
int sum=tr[r]-tr[l-1];
cout<<sum<<endl;
}
}
二维前缀和
int a[N][N];
int tr[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++)
{
scanf("%d",&a[i][j]);
tr[i][j]=tr[i-1][j]+tr[i][j-1]-tr[i-1][j-1]+a[i][j];
}
while(q--)
{
int x1,x2,y1,y2;
cin>>x1>>y1>>x2>>y2;
int sum=tr[x2][y2]-tr[x2][y1-1]-tr[x1-1][y2]+tr[x1-1][y1-1];
cout<<sum<<endl;
}
}
差分:
一维差分
int a[N];
int b[N];
void insert(int l,int r,int t)
{
b[l]+=t;
b[r+1]-=t;
}
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++)
insert(i,i,a[i]);
while(m--)
{
int l,r,t;
cin>>l>>r>>t;
insert(l,r,t);
}
for(int i=1;i<=n;i++)
a[i]=a[i-1]+b[i];
for(int i=1;i<=n;i++)
cout<<a[i]<<" ";
}
二维差分
const int N = 1010;
int a[N][N];
int tr[N][N];
void insert(int x1,int y1,int x2,int y2,int c)
{
tr[x1][y1]+=c;
tr[x2+1][y2+1]+=c;
tr[x1][y2+1]-=c;
tr[x2+1][y1]-=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++)
scanf("%d",&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,x2,y1,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]+tr[i][j];
for(int i=1; i<=n; i++)
{
for(int j=1; j<=m; j++)
cout<<a[i][j]<<" ";
cout<<endl;
}
}