牛客练习赛50 B tokitsukaze and Hash Table （并查集找最近空位）

大致题意

要依次插入n个数字，如果 ai %n 的位置空着，就放在 ai%n ，否则就一直向右找到n-1 ,如果还没有就从 0 向右继续找。

思路

用并查集维护每个位置右边最近的空位，初始 king[x]=x 。如果某个位置 x 被占用，那么 king[x]=(x+1)%n;

代码

#include
using namespace std;
#define maxn 1000005
#define maxm 1000006
#define ll long long int
#define INF 0x3f3f3f3f
#define inc(i,l,r) for(int i=l;i<=r;i++)
#define dec(i,r,l) for(int i=r;i>=l;i--)
#define mem(a) memset(a,0,sizeof(a))
#define sqr(x) (x*x)
#define inf (ll)2e18+1
int read(){
    int x=0,f=1;char ch=getchar();
    while(!isdigit(ch)){if(ch=='-')f=-1;ch=getchar();}
     while(isdigit(ch))x=x*10+ch-'0',ch=getchar();
     return f*x;
}
int n,x,a[maxn];
int king[maxn];
int getfa(int x){return king[x]==x?x:king[x]=getfa(king[x]);}
int main()
{
    n=read();
    inc(i,0,n-1)king[i]=i;
    inc(i,1,n){
        x=read();
        int pos=getfa(x%n);
        a[pos]=x;
        king[pos]=(pos+1)%n;
    }
    inc(i,0,n-1)printf("%d%c",a[i]," \n"[i==n-1]);
    return 0;
}