BZOJ 2054 疯狂的馒头

并查集把染过色的并在一起。倒着染色。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define maxn 10000050
using namespace std;
int n,m,l[maxn],r[maxn],father[maxn/],p,q,col[maxn/];
int getfather(int x)
{
    if (x!=father[x]) father[x]=getfather(father[x]);
    return father[x];
}
void unionn(int x,int y)
{
    if ((!col[x]) && (!col[y])) return;
    int f1=getfather(x),f2=getfather(y);
    father[f1]=f2;
}
int main()
{
    scanf("%d%d%d%d",&n,&m,&p,&q);
    for (int i=;i<=m;i++)
    {
        l[i]=(i*p+q)%n+;r[i]=(i*q+p)%n+;
        if (l[i]>r[i]) swap(l[i],r[i]);
    }
    for (int i=;i<=n;i++) father[i]=i;
    for (int i=m;i>=;i--)
    {
        int now=l[i];
        if (col[now]) now=getfather(now)+;
        while (now<=r[i])
        {
            unionn(now-,now);unionn(now,now+);
            col[now]=i;
            now++;
            if (col[now]) now=getfather(now)+;
        }
    }
    for (int i=;i<=n;i++) printf("%d\n",col[i]);
    return ;
}