BZOJ2054 疯狂的馒头

Link
并查集维护染色模板题。
倒序染色，用并查集维护下一个没有被染色的点。
如果用路径压缩+按秩合并可以做到\(O(n\alpha(n))\)。
反正复杂度瓶颈在读入上所以基本上都是以某种方式生成数据。

#include<bits/stdc++.h>
using namespace std;
namespace IO
{
    char ibuf[(1<<21)+1],obuf[(1<<21)+1],st[15],*iS,*iT,*oS=obuf,*oT=obuf+(1<<21);
    char Get(){return (iS==iT? (iT=(iS=ibuf)+fread(ibuf,1,(1<<21)+1,stdin),(iS==iT? EOF:*iS++)):*iS++);}
    void Flush(){fwrite(obuf,1,oS-obuf,stdout),oS=obuf;}
    void Put(char x){*oS++=x;if(oS==oT)Flush();}
    int read(){int x=0;char ch=Get();while(ch>57||ch<48)ch=Get();while(ch>=48&&ch<=57)x=x*10+(ch^48),ch=Get();return x;}
    void write(int x){int top=0;if(!x)Put('0');while(x)st[++top]=(x%10)+48,x/=10;while(top)Put(st[top--]);Put('\n');}
}
using namespace IO;
const int N=1000007;
int fa[N],col[N];
int Find(int x){return x==fa[x]? x:fa[x]=Find(fa[x]);}
int main()
{
    int n=read(),m=read(),p=read(),q=read(),l,r,i,j;
    for(i=1;i<=n+1;++i) fa[i]=i;
    while(m)
    {
	l=(1ll*m*p+q)%n+1,r=(1ll*m*q+p)%n+1;
	if(l>r) swap(l,r);
	for(i=Find(l);i<=r;i=j) col[i]=m,fa[i]=j=Find(i+1);
	--m;
    }
    for(i=1;i<=n;++i) write(col[i]);
    return Flush(),0;
}