「题解」BZOJ #2054. 疯狂的馒头

题目

#2054. 疯狂的馒头

简化题意

给序列的区间染色（一开始都是 \(0\)），问最后序列是什么颜色的。

思路

并查集。

倒序染色这样就不用去考虑已经染过色的了染过色的用并查集维护。

坑点：题目中给你的左端点不一定小于等于右端点。

Code

#include <cstdio>
#include <cstring>
#include <string>
#include <iostream>
#include <algorithm>
#define MAXN 1000001

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

int find(int x) { return fa[x] == x ? x : fa[x] = find(fa[x]); }

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) {
        int l = (1ll * i * p + 1ll * q) % n + 1, r = (1ll * i * q + 1ll* p) % n + 1;
        if (l > r) std::swap(l, r);
        for (int j = r; j >= l;) {
            if (fa[j] == j) {
                col[j] = i;
                fa[j] = fa[j - 1];
            }
            j = find(fa[j]);
        }
    }
    for (int i = 1; i <= n; ++i) printf("%d\n", col[i]);
    return 0;
}