【BZOJ 2054】 2054: 疯狂的馒头 (并查集特技)

Input

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

Output

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

Sample Input

4 3 2 4

Sample Output

2
2
3
0

HINT

 

 

 

【分析】

　　想要不带log，就从后往前做，每个点只染色一次，用神奇的并查集维护即可。

 

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
#define Maxn 1000010

int col[Maxn],rt[Maxn];

int rtt(int x)
{
    if(rt[x]!=x) rt[x]=rtt(rt[x]);
    return rt[x];
}

int tot=0;
void solve(int l,int r,int nw)
{
    while(l<=r)
    {
        if(col[l]==0)
        {
            col[l]=nw;
            rt[l]=rtt(l+1);
            l=rtt(l);
            tot++;
        }
        l=rtt(l);
    }
}

int main()
{
    int n,m,p,q;
    scanf("%d%d%d%d",&n,&m,&p,&q);
    memset(col,0,sizeof(col));
    for(int i=1;i<=n+1;i++) rt[i]=i;
    for(int i=m;i>=1;i--)
    {
        if(tot==n) break;
        int l=(i*p+q)%n+1,r=(i*q+p)%n+1;
        if(l<=r) solve(l,r,i);
        else solve(r,l,i);
    }
    for(int i=1;i<=n;i++) printf("%d
",col[i]);
    return 0;
}

还有就是不要理解错那个区间。。[r,l]也是可以的。

2017-03-06 20:57:21