密码游戏 - bzoj十连测 Test7 T2

密码游戏 - bzoj十连测 Test7 T2

并查集的神奇应用

题解：

一开始我和hzw都认为a可以随便取。。。

————————zz分割线————————

$b[(a[(x[i]+a_0)\%m]+b_0)\%m]=y[i]$

方便看把%m省去：

$b[a[x[i]+a_0]+b_0]=y[i]$

$a_0=i\%m,b_0=i/m\%m$

设$y[i]=y[j]$

则$b[a[x[i]+a_1]+b_1]=b[a[x[i]+a_2]+b_2]$

可以整理成$a[p1]-a[p2]=k$的形式

然后用权值并查集去维护样的式子，就算出a[]来啦

然后根据a[]推b[]

太神啦QwQ

Code：

#include <iostream>
#include <cstring>
#include <cstdio>
#define D(x) cout<<#x<<" = "<<x<<"  "
#define E cout<<endl
using namespace std;
const int N = 200005;

int n,m,X[N],Y[N],a[N],b[N],pa[N],dic[N]; 

int find(int x){ 
    if(x!=pa[x]){
        int f=find(pa[x]);
//      D(x); D(pa[x]); D(a[x]); D(a[pa[x]]); E;
        a[x]=(a[x]+a[pa[x]])%m;
        pa[x]=f;
    }
    return pa[x];
}
void merge(int x,int y,int k){
    int xx=find(x), yy=find(y);
    if(xx!=yy){
//      D(x); D(y); D(k); E;
//      D(xx); D(yy); E;
//      D(a[x]); D(a[y]); E;
        pa[xx]=yy; a[xx]=(k-a[x]+a[y])%m;
//      D(a[xx]); E;
    }
}

int main(){
    freopen("password.in","r",stdin);
    freopen("password.out","w",stdout);
    scanf("%d%d",&n,&m);
    for(int i=0;i<n;i++) scanf("%d",X+i);
    for(int i=0;i<n;i++) scanf("%d",Y+i);
    memset(dic,-1,sizeof(dic)); 
    for(int i=0;i<m;i++) pa[i]=i;
    for(int i=0;i<n;i++){
//      D(i); E;
        if(dic[Y[i]]==-1){
            dic[Y[i]]=i;
        }
        else{
            int j=dic[Y[i]];
            int a1=i%m, b1=(i/m)%m;
            int a2=j%m, b2=(j/m)%m;
            int p1=(a1+X[i])%m, p2=(a2+X[j])%m;
            int k=(b2-b1)%m;
            merge(p1,p2,k);
        }
    }
    for(int i=0;i<m;i++){
        find(i); a[i]=(a[i]%m+m)%m;
    }
    for(int i=0;i<n;i++){
        int a0=i%m, b0=(i/m)%m;
        b[(a[(X[i]+a0)%m]+b0)%m]=Y[i];
    }
    for(int i=0;i<m;i++) printf("%d ",a[i]); puts("");
    for(int i=0;i<m;i++) printf("%d ",b[i]); puts("");
}