[BZOJ4538]网络

题目链接：BZOJ4538

题目大意
给出一颗树，要求支持三种操作：
1. 声明一个经过树上两点之间路径的带权任务；
2. 取消$t$时刻的任务；
3. 询问不经过$x$点的未取消任务中权值最大值。

分析
1. 乍一看树剖，其实就是树剖+线段树+堆。
2. 首先我们把操作1取反，即在路径的补集上加任务，那么操作3就变成了所有经过$x$的未取消任务的最大值。
3. 线段树每个节点套一个堆，代表覆盖这整个区间的任务的集合；若任务不足以覆盖整个区间，处理方式和经典线段树一样；询问时询问从线段树第一层一直到最后一层$x$点的$max$；
4. 求补集的话，先把原集合在线段树上的各个左右端点$mark[i]$求出，然后加上端点$0$和端点$n+1$排一遍序，区间$[mark[i]+1,mark[i+1]-1]$就是补集了，复杂度$O(logNlog(logN))$。

上代码

#include <queue>
#include <vector>
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;

const int N = 1e5 + 10;
const int M = 2e5 + 10;

int n, m;
inline int read() {
    char ch;
    int ans = 0, neg = 1;
    while (ch = getchar(), ch < '0' || ch > '9')
        if (ch == '-') neg  = -1;
    while (ch >= '0' && ch <= '9')
        ans = ans * 10 + ch - '0', ch = getchar();
    return ans * neg;
}

bool book[M];
vector <int> edge[N];

namespace HLD {
    struct Poi { // poi？ poi！
        int id, val;
        Poi() {}
        Poi(int a, int b) : id(a), val(b) {}
        inline bool operator < (const Poi &a) const {
            return val < a.val;
        }
    }; typedef priority_queue <Poi> HP;
    int cntSeg;
    #define lc(a) (a << 1)
    #define rc(a) (lc(a) | 1)
    struct SegTree {
        HP H;
        int l, r;
    } T[N * 3];

    int fa[N], son[N], size[N], dep[N];
    int top[N], plc[N], replc[N];
    void dfs1(int a, int par) {
        fa[a] = par, size[a] = 1, dep[a] = dep[par] + 1;
        for (int i = 0; i < edge[a].size(); i++) {
            int tmp = edge[a][i];
            if (tmp == par) continue;
            dfs1(tmp, a), size[a] += size[tmp];
            if (size[tmp] > size[son[a]]) son[a] = tmp;
        }
    }
    void dfs2(int a, int up) {
        top[a] = up, replc[plc[a] = ++cntSeg] = a;
        if (son[a]) dfs2(son[a], up);
        for (int i = 0; i < edge[a].size(); i++) {
            int tmp = edge[a][i];
            if (tmp == fa[a] || tmp == son[a]) continue;
            dfs2(tmp, tmp);
        }
    }
    void build(int a, int l, int r) {
        T[a].l = l, T[a].r = r;
        if (l == r) return;
        int mid = (l + r) >> 1;
        build(lc(a), l, mid), build(rc(a), mid + 1, r);
    }
    Poi mdf;
    void modify(int a, int l, int r) {
        if (l > r) return;
        if (T[a].l == l && T[a].r == r) {
            T[a].H.push(mdf); return;
        }
        int mid = (T[a].l + T[a].r) >> 1;
        if (r <= mid) modify(lc(a), l, r);
        else if (l > mid) modify(rc(a), l, r);
        else modify(lc(a), l, mid), modify(rc(a), mid + 1, r);
    }
    int getMax(int a, int p) {
        while (!T[a].H.empty() && book[T[a].H.top().id]) T[a].H.pop();
        int tmp =  T[a].H.empty() ? -1 : T[a].H.top().val;
        if (T[a].l == p && T[a].r == p) return tmp;
        int mid = (T[a].l + T[a].r) >> 1;
        if (p > mid) return max(tmp, getMax(rc(a), p));
        else return max(tmp, getMax(lc(a), p));
    }
    int mark[N];
    void decom(int a, int b, int c, int d) {
        int cnt = 0; mdf = Poi(d, c);
        while (top[a] != top[b]) {
            if (dep[top[a]] < dep[top[b]]) swap(a, b);
            mark[++cnt] = plc[top[a]], mark[++cnt] = plc[a];
            a = fa[top[a]];
        }
        if (dep[a] < dep[b]) swap(a, b);
        mark[++cnt] = plc[b], mark[++cnt] = plc[a];
        mark[++cnt] = 0, mark[++cnt] = n + 1;
        sort(mark + 1, mark + cnt + 1);
        for (int i = 2; i <= cnt; i += 2)
            modify(1, mark[i - 1] + 1, mark[i] - 1);
    }
}

void init() {
    n = read(), m = read();
    for (int i = 1; i < n; i++) {
        int a = read(), b = read();
        edge[a].push_back(b), edge[b].push_back(a);
    }
    HLD::dfs1(1, 0), HLD::dfs2(1, 1), HLD::build(1, 1, n);
}
void figure() {
    int a, b, c;
    for (int i = 1; i <= m; i++) {
        switch (read()) {
            case 0:
                a = read(), b = read(), c = read();
                HLD::decom(a, b, c, i); break;
            case 1:
                book[read()] = true; break;
            default:
                a = HLD::plc[read()];
                printf("%d\n", HLD::getMax(1, a)); break;
        }
    }
}

int main() {
    init();
    figure();
    return 0;
}

以上