CF600E Lomsat gelral （线段树合并）

相当于是线段树合并的模板题，比（雨天的尾巴）还要板。

唯一注意的是线段树的更新，因为同一子树中可能有多种颜色占主导地位，要输出编号和，比如一颗子树中，1出现3次（最多），3出现3次，那么应该输出4。

#include<bits/stdc++.h>
#define ll long long
using namespace std;
const int N = 1e5 + 10, M = 1e5;
struct node {
    int lc, rc;
    ll dat, num;//dat次数最多颜色的次数，num编号和 
}tr[N * 50];
int head[N], to[N << 1], nxt[N << 1], tot;
int n, rt[N], cnt, co[N];
ll ans[N];
void add(int x, int y) {
    nxt[++tot] = head[x];
    head[x] = tot;
    to[tot] = y;
}

void pushup(int k) {//注意
    node a = tr[tr[k].lc], b = tr[tr[k].rc];
    if (a.dat > b.dat) {
        tr[k].dat = a.dat;
        tr[k].num = a.num;
    }
    else if (a.dat < b.dat) {
        tr[k].dat = b.dat;
        tr[k].num = b.num;
    }
    else {
        tr[k].dat = a.dat;
        tr[k].num = a.num + b.num;
    }
}

void insert(int p, int l, int r, int pos, int val) {
    if (l == r) {
        tr[p].dat += val;
        tr[p].num = l;
        return ;
    }
    int mid = (l + r) >> 1;
    if (pos <= mid) {
        if (!tr[p].lc) tr[p].lc = ++cnt;
        insert(tr[p].lc, l, mid, pos, val);
    }
    else {
        if (!tr[p].rc) tr[p].rc = ++cnt;
        insert(tr[p].rc, mid + 1, r, pos, val);
    }
    pushup(p);
} 

int merge(int p, int q, int l, int r) {
    if (!p || !q) return p + q;
    if (l == r) {
        tr[p].dat += tr[q].dat;
        tr[p].num = l;
        return p;
    }
    int mid = (l + r) >> 1;
    tr[p].lc = merge(tr[p].lc, tr[q].lc, l, mid);
    tr[p].rc = merge(tr[p].rc, tr[q].rc, mid + 1, r);
    pushup(p);
    return p;
}

void dfs(int u, int f) {
    for (int i = head[u]; i; i = nxt[i]) {
        int v = to[i];
        if (v == f) continue;
        dfs(v, u);
        rt[u] = merge(rt[u], rt[v], 1, M);
    }
    ans[u] = tr[rt[u]].num;
}

int main() {
    scanf("%d", &n);
    for (int i = 1; i <= n; i++) {
        scanf("%d", &co[i]);
        rt[i] = ++cnt;
        insert(rt[i], 1, M, co[i], 1);
    }
    for (int i = 1; i < n; i++) {
        int a, b;
        scanf("%d%d", &a, &b);
        add(a, b),add(b, a);
    }
    dfs(1, 0);
    for (int i = 1; i <= n; i++) printf("%lld ", ans[i]);
    return 0;
}
/*
4
1 2 3 4
1 2
2 3
2 4
*/

 

---

如果您觉得阅读本文对您有帮助，请点一下“推荐”按钮，您的“推荐”将是我最大的写作动力！欢迎各位转载，但是未经作者本人同意，转载文章之后必须在文章页面明显位置给出作者和原文连接，否则保留追究法律责任的权利。