【玄关】15pts求调
查看原帖
【玄关】15pts求调
528430
FiraCode楼主2024/1/21 12:12

就是直接线段树合并。

#include <bits/stdc++.h>

using namespace std;

const int N = 100010;

int n, m;
vector<int> e[N];

int idx;
int rt[N];
int ans[N];
struct Node {
    int lson, rson;
    int maxn, col;
} tr[N * 80];

void pushup(int u) {
    if (tr[tr[u].lson].maxn > tr[tr[u].rson].maxn) {
        tr[u].maxn = tr[tr[u].lson].maxn;
        tr[u].col = tr[tr[u].lson].col;
    } else if (tr[tr[u].lson].maxn < tr[tr[u].rson].maxn) {
        tr[u].maxn = tr[tr[u].rson].maxn;
        tr[u].col = tr[tr[u].rson].col;
    } else {
        if (tr[tr[u].lson].col < tr[tr[u].rson].col) {
            tr[u].maxn = tr[tr[u].lson].maxn;
            tr[u].col = tr[tr[u].lson].col;
        } else {
            tr[u].maxn = tr[tr[u].rson].maxn;
            tr[u].col = tr[tr[u].rson].col;
        }
    }
}

void modify(int &u, int l, int r, int d, int x) {
    if (!u) u = ++idx;
    if (l == r) {
        tr[u].maxn += x;
        tr[u].col = l;
        return;
    }

    int mid = (l + r) >> 1;

    if (d <= mid) modify(tr[u].lson, l, mid, d, x);
    else modify(tr[u].rson, mid + 1, r, d, x);

    pushup(u);
}

int merge(int u, int v, int l, int r) {
    if (!u) return v;
    if (!v) return u;

    if (l == r) {
        tr[u].maxn += tr[v].maxn;
        tr[u].col = l;
        return u;
    }

    int mid = (l + r) >> 1;

    tr[u].lson = merge(tr[u].lson, tr[v].lson, l, mid);
    tr[u].rson = merge(tr[u].rson, tr[v].rson, mid + 1, r);

    pushup(u);

    return u;
}

void solve(int u, int fa = -1) {
    for (auto v : e[u]) {
        if (v == fa) continue;

        solve(v, u);

        merge(rt[u], rt[v], 1, 100000);
    }

    ans[u] = tr[rt[u]].col;
}

int idx_dfn;
int fa[N], dep[N], top[N], son[N], sz[N], dfn[N];

void dfs1(int u, int f = 0) {
    dep[u] = dep[f] + 1;
    sz[u] = 1;
    fa[u] = f;
 
    for (auto v : e[u]) {
        if (v == f) continue;

        dfs1(v, u);

        sz[u] += sz[v];
        if (sz[son[u]] < sz[v]) son[u] = v;
    }
}

void dfs2(int u, int topf) {
    dfn[u] = ++idx_dfn;
    top[u] = topf;

    if (son[u])
        dfs2(son[u], topf);

    for (auto v : e[u]) if (v != son[u] && v != fa[u])
        dfs2(v, v);
}

int LCA(int u, int v) {
    while (top[u] != top[v]) {
        if (dep[top[u]] < dep[top[v]]) swap(u, v);
        u = fa[top[u]];
    }

    if (dep[u] > dep[v]) swap(u, v);

    return u;
}

int main() {
    scanf("%d%d", &n, &m);
    for (int i = 1; i < n; ++i) {
        int u, v;
        scanf("%d%d", &u, &v);
        e[u].push_back(v);
        e[v].push_back(u);
        rt[i] = i;
    }
    idx = n;

    dfs1(1, 0), dfs2(1, 1);

    for (int i = 1; i <= m; ++i) {
        int u, v, w;
        scanf("%d%d%d", &u, &v, &w);
        modify(rt[u], 1, 100000, w, 1);
        modify(rt[v], 1, 100000, w, 1);
        // cout << LCA(u, v) << "---\n";
        if (LCA(u, v) != 1)
            modify(rt[fa[LCA(u, v)]], 1, 100000, w, -2);
    }

    solve(1);

    for (int i = 1; i <= n; ++i) printf("%d\n", ans[i]);

    return 0;
}
2024/1/21 12:12
加载中...