CF911G Mass Change Queries【动态开点线段树】

题意

给出一个数列,有$q$个操作,每种操作是把区间$[l,r]$中等于$x$的数改成$y$.输出$q$步操作完的数列.

思路

由于数列中每个数的大小只有100，可以对每一个数建立一颗权值线段树，也就是建一百颗权值线段树。
当然这样会导致MLE，于是考虑动态开点线段树。
考虑每一个操作$(l,r,x,y)$，需要把$x$权值线段树中的$(l,r)$区间有值的位置清空，把$y$权值线段树中对应的位置赋值。
那么其实可以使用类似线段树合并的操作，把$x$中有权值的区间（树的枝节）合并到$y$的权值线段树中。

代码

#include <iostream>
#include <algorithm>
#include <cstring>
#include <cstdio>
#include <set>
#include <vector>
#include <map>
#include <queue>
#include <cmath>

using namespace std;

typedef long long LL;
typedef pair<int, int> PII;

const int N = 200005, M = 20000005;

struct TreeNode
{
    int l, r;
}t[M];
int rt[105], id;
int ans[N];

void make(int& i, int l, int r)
{
    if (i) return;
    i = ++ id;
}

void updatePos(int& i, int l, int r, int pos)
{
    make(i, l, r);
    if (l == r) return;
    int mid = (l + r) >> 1;
    if (pos <= mid) updatePos(t[i].l, l, mid, pos);
    else updatePos(t[i].r, mid + 1, r, pos);
}

int Merge(int i, int j)
{
    if (!i || !j) return i | j;
    t[i].l = Merge(t[i].l, t[j].l);
    t[i].r = Merge(t[i].r, t[j].r);
    return i;
}

void updateRange(int& i, int& j, int l, int r, int L, int R)
{
    if (!i) return;
    if (L <= l && r <= R)
    {
        j = Merge(i, j);
        i = 0;
        return;
    }
    make(j, l, r);
    int mid = (l + r) >> 1;
    if (L <= mid) updateRange(t[i].l, t[j].l, l, mid, L, R);
    if (R > mid) updateRange(t[i].r, t[j].r, mid + 1, r, L, R);
}

void solve(int i, int l, int r, int k)
{
    if (!i) return;
    if (l == r)
    {
        ans[l] = k;
        return;
    }
    int mid = (l + r) >> 1;
    solve(t[i].l, l, mid, k);
    solve(t[i].r, mid + 1, r, k);
}

int main()
{
    #ifdef ZYCMH
    freopen("1.in", "r", stdin);
    freopen("1.out", "w", stdout);
    #endif
    int n;
    scanf("%d", &n);
    for (int i = 1; i <= n; i ++ )
    {
        int x;
        scanf("%d", &x);
        updatePos(rt[x], 1, n, i);
    }

    int m;
    scanf("%d", &m);
    for (int i = 1; i <= m; i ++ )
    {
        int l, r, x, y;
        scanf("%d%d%d%d", &l, &r, &x, &y);
        if (x == y) continue;
        updateRange(rt[x], rt[y], 1, n, l, r);
    }

    for (int i = 1; i <= 100; i ++ )
        solve(rt[i], 1, n, i);
    
    for (int i = 1; i <= n; i ++ )
        printf("%d ", ans[i]);
    puts("");

    return 0;
}