P2391 白雪皑皑 / BZOJ 2054 疯狂的馒头

有\(n\)个数和\(m\)个操作，\(n\)个数初始为\(0\)，每次给定\(l_i,r_i,c_i\)，要将\([l_i,r_i]\)中的数变为\(c_i\)，输出最后的序列，\(n,m \le 10^6\)。

还是建议看一下原题

并查集经典题。

显然一个点的颜色取决于它最后一次被染的颜色，那么我们考虑反向染色。（指操作全部反过来）

并查集维护颜色块，向它右边的点连边。

注意：\(f\)预处理时处理到\(n + 1\)，否则当染到\(n\)时\(f_n = n + 1\)，而\(f_{n + 1} = 0\)，递归爆栈就没了。悲

#include <bits/stdc++.h>
using namespace std;

const int N = 1e6 + 5;

int n, m, p, q;

int f[N];
int a[N];

int find(int x)
{
    if (x != f[x])
        f[x] = find(f[x]);
    return f[x];
}

int main(void)
{
    scanf("%d%d%d%d", &n, &m, &p, &q);
    for (int i = 1; i <= n + 1; i++)
        f[i] = i;
    for (int i = m; i >= 1; i--)
    {
        int l, r;
        l = (i * p + q) % n + 1;
        r = (i * q + p) % n + 1;
        if (l > r)
            swap(l, r);
        int col = i;
        int fl = find(l);
        for (int j = fl; j <= r; j = find(j))
        {
            if (find(j) == j)
            {
                a[j] = col;
                f[j] = j + 1;
            }
        }
    }
    for (int i = 1; i <= n; i++)
    {
        printf("%d\n", a[i]);
    }
    return 0;
}