好精妙的一道题啊
传送门:here
大致题意:有一个$ n*m$的矩阵,q次询问每次交换给定两个无交矩阵的对应元素,求操作后的最终矩阵?
数据范围:$ n,m<=1000, q<=10000$
模拟赛的时候考到类似的,当时很快想到对每行维护一个$ treap/splay$的优秀做法,
然后感觉复杂度一千万$ *log$甚至比暴力慢(事实证明确实如此)
正解其实又好写又快速,不过对我这种数据结构做傻的菜鸡来说考场上确实是想不到啊....
$ solution:$
对于每个元素维护两个指针R(右边的那个元素标号)以及D(下边的那个元素标号)还有这个点的元素值
注意元素从$ (0,0)$开始记录
对于每次修改,会发现只有矩阵边缘的点的指针需要被修改(如图所示)
交换红蓝两个矩形时,只有涂黄色的格子下指针需要修改,涂绿色的格子右指针需要修改,对应swap一下即可
容易发现每次修改只有$ 2n+2m$格被修改了,因此总复杂度为$ O((n+m)q)$,非常优秀
(注意存图的时候第一行和第一列要空出来给空元素)
代码理解之后非常好写,以下是我的代码
#include<cstdio>
#include<iostream>
#define rt register int
using namespace std;
int i,j,k,m,n,x,y,z,cnt,p;
struct node{
int v,D,R;//D表示下,R表示右
}a[];
int main(){
scanf("%d %d %d",&n,&m,&k);n++;m++;
for(rt i=;i<=n;i++)
for(rt j=;j<=m;j++){
int v=(i-)*m+j;
if(i>&&j>)scanf("%d",&a[v].v);
if(i!=n)a[v].D=v+m;
if(j!=m)a[v].R=v+;
}
while(k--){
int X1,Y1,X2,Y2,L,C,g1=,g2=,g3,g4;
scanf("%d %d %d %d %d %d",&X1,&Y1,&X2,&Y2,&L,&C);
while(--X1)g1=a[g1].D;while(--X2)g2=a[g2].D;
while(--Y1)g1=a[g1].R;while(--Y2)g2=a[g2].R;
g3=g1;g4=g2;
for(rt i=;i<=L;i++)g1=a[g1].D,g2=a[g2].D,swap(a[g1].R,a[g2].R);
for(rt i=;i<=C;i++)g1=a[g1].R,g2=a[g2].R,swap(a[g1].D,a[g2].D);
for(rt i=;i<=C;i++)g3=a[g3].R,g4=a[g4].R,swap(a[g3].D,a[g4].D);
for(rt i=;i<=L;i++)g3=a[g3].D,g4=a[g4].D,swap(a[g3].R,a[g4].R);
}
int pla=a[a[].D].R;
for(rt i=;i<n;i++,pla=a[pla].D){
for(rt j=,d=pla;j<m;j++,d=a[d].R)printf("%d ",a[d].v);
putchar('\n');
}
return ;
}