疯狂的馒头
Time Limit: 10 Sec Memory Limit: 162 MB
Description
Input
第一行四个正整数N，M，p，q
Output
一共输出N行，第i行表示第i个馒头的最终颜色（如果最终颜色是白色就输出0）。
Sample Input
4 3 2 4
Sample Output
2
2
3
0
HINT
这道题介绍了并查集一种比较有代表性的姿势。。。
由于这个涂色覆盖的问题，我们直接倒着涂就好了，就变成涂完就不覆盖了。。。。
然后我们再用并查集来维护这些颜色。。。
这里我们用fa数组来记录下一个没有被涂过色的是哪个。。。。
ok
#include<bits/stdc++.h>
using namespace std;
const int maxn = 1e6 + 5;
int n, m, p, q, x, y;
int fa[maxn], ans[maxn];
int find(int t)
{
if(t > n) return fa[t] = t;
if(t == fa[t]) return t;
return fa[t] = find(fa[t]);
}
int main()
{
scanf("%d%d%d%d", &n, &m, &p, &q);
for(int i = 1; i <= n; ++i) fa[i] = i;
for(int i = m; i >= 1; --i){
x = ((long long)i * p + q) % n + 1; y = ((long long)i * q + p) % n + 1;
if(x > y) swap(x, y);
for(int j = find(x); j <= y; j = find(j)){
ans[j] = i; fa[j] = j + 1;
}
}
for(int i = 1; i <= n; ++i) printf("%d\n", ans[i]);
return 0;
}