[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;}

以上