【bzoj2054】【疯狂的馒头】【并查集】

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 <iostream>
#include <algorithm>
#define M 1001001
using namespace std;
int n,m,p,q;
int a[M];
namespace Union_Find_Set{
    int fa[M];
    int Find(int x)
    {
        if(!fa[x]||fa[x]==x)
            return fa[x]=x;
        return fa[x]=Find(fa[x]);
    }
}
int main()
{
    using namespace Union_Find_Set;
    int i,j;
    cin>>n>>m>>p>>q;
    for(i=m;i;i--)
    {
        int x=((long long)i*p+q)%n+1;
        int y=((long long)i*q+p)%n+1;
        if(x>y) swap(x,y);
        for(j=Find(x);j<=y;j=Find(j))
            a[j]=i,fa[j]=j+1;
    }
    for(i=1;i<=n;i++)
        printf("%d\n",a[i]);
    return 0;
}