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