BZOJ2054 疯狂的馒头 并查集

2054: 疯狂的馒头

Time Limit: 10 Sec Memory Limit: 162 MB
Submit: 974 Solved: 423
[Submit][Status][Discuss]
Description

Input

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

Output

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

Sample Input

4 3 2 4
Sample Output

2

2

3

0
HINT

分析：
也是并查集的一个典型应用啊23333特别巧。
考虑每个馒头只有最后一次染色有效，所以可以倒着染色，对于已经染过色的馒头就跳过。
关于“跳过”这一步，我们可以用跳表（应该是叫这个），也可以用并查集。每次对这个馒头进行染色之后，就把它的父亲变为后面一个，（如果后一个已经染过色，那么它的父亲也会改变），这样最后找到的祖先即是一个没有染过色的馒头。
这样就可以直接找到下一个没有染过色的馒头，进行染色。
总的时间复杂度为O(n+m)
注意附初始值的时候要到n+1，因为n会和n+1合并。（这个贡献了2发RE）

/**************************************************************
    Problem: 2054
    User: LaLaLa112138
    Language: C++
    Result: Accepted
    Time:2568 ms
    Memory:110112 kb
****************************************************************/

//BZOJ 2054
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;

const int N = 10000010;

int n,m,p,q;
int col[N];
int fa[N];

int getfather(int x){
    if(fa[x]==x) return x;
    else return fa[x]=getfather(fa[x]);
}

int main(){
    scanf("%d%d%d%d",&n,&m,&p,&q);
    for(register int i=1;i<=n+1;i++) fa[i]=i;
    for(register int i=m;i>=1;i--){
        int u=(i*p+q)%n+1,v=(i*q+p)%n+1;
        if(u>v) swap(u,v);
        for(int j=getfather(u);j<=v;j=getfather(j)){
            col[j]=i;fa[j]=j+1;
        }
    }
    for(register int i=1;i<=n;i++) printf("%d\n",col[i]);
    return 0;
}