bzoj2054 疯狂的馒头

疯狂的馒头

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
", ans[i]);
	return 0;
}