bzoj 2054 疯狂的馒头

想到了要用一种东西维护下一个没被染色的是什么东西，但是没想到是并查集。这道题就相当于在每一个集合里的数都会跳到一个相同的点，就相当于并查集的代表元素。

如果将一个点染色，就把它的父亲设为i+1，这样并查集的代表元素就是下一个没有被染色的点。

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<algorithm>

#define ll long long
#define inf 1e9
#define eps 1e-10
#define md
#define N 1000010
using namespace std;
int fa[N],col[N];
int find(int x) { return fa[x]==x?x:fa[x]=find(fa[x]);}
int main()
{
	ll n,m,p,q;
	scanf("%lld%lld%lld%lld",&n,&m,&p,&q);
	for (int i=1;i<=n+1;i++) fa[i]=i;
	for (int i=m;i;i--)
	{
		int l=(i*p+q)%n+1,r=(i*q+p)%n+1;
		if (l>r) swap(l,r);
		for (int x=find(l);x<=r;x=find(x))
		{
			col[x]=i;
			fa[x]=x+1;
		}
	}
	for (int i=1;i<=n;i++) printf("%d\n",col[i]);
	return 0;
}