BZOJ 2054 并查集 解题报告

2054: 疯狂的馒头

Description

Input

第一行四个正整数N，M，p，q

Output

一共输出N行，第i行表示第i个馒头的最终颜色（如果最终颜色是白色就输出0）。

Sample Input

4 3 2 4

Sample Output

2
2
3
0

HINT

【解题报告】
因为只询问最后的颜色，所以将操作倒着处理就可以了，不用管前面怎么染的。
用并查集维护每个馒头后面第一个白馒头(包括自己)，然后每次染色暴力染区间内所有空馒头，同时把染了色的馒头的集合和后面一个的集合进行合并。

代码如下：

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
#define N 1000005
#define M 10000005

int n,m,p,q,a[N],f[N];

int find(int x)
{
    return f[x]==x?x:f[x]=find(f[x]);
}
int main()
{
    scanf("%d%d%d%d",&n,&m,&p,&q);
    for(int i=1;i<=n+1;++i) f[i]=i;
    for(int i=m;i>=1;--i)
    {
        int x=(i*p+q)%n+1,y=(i*q+p)%n+1;
        if(x>y) swap(x,y);
        for(int j=find(x);j<=y;j=find(j)) a[j]=i,f[j]=j+1;
    }
    for(int i=1;i<=n;++i) printf("%d\n",a[i]);
    return 0;
}