CodeForces - 600E Lomsat gelral（线段树合并）

题目链接

线段树的合并是从这里学的

题意 ：给你一棵以$1$为根节点的树，树上每个点有一个颜色，用$1$~$n$的编号表示，问在以每个点为根节点的子树中，出现次数最多（可能有多种颜色出现次数一样）的颜色的编号和是多少。

题解：处理出现数量问题，并且大的子树的值可以由它的左右子树的值确定，所以想到了线段树的合并。空间复杂度在玄学的证明下大概是$O(nlogn)$，但是好像会带个2的常数，所以要开$2nlogn$倍。$1e5$大概要开$32$倍再大一点。

#include <bits/stdc++.h>
using namespace std;
#define lson t[id].l
#define rson t[id].r

typedef long long ll;

template <class T> inline void read(T &x) {
    int f = 0; x = 0; char ch = getchar();
    for (; !isdigit(ch); ch = getchar()) f |= (ch == '-');
    for (; isdigit(ch); ch = getchar()) x = x * 10 + ch - '0';
    if (f) x = -x;
}
const int maxn = 1e5 + 7;

struct Seg{
    int l, r，num;
    ll sum;
}t[maxn * 50];
int n, tot, cor[maxn], rt[maxn];
ll ans[maxn];
vector<int> G[maxn];

void up(int id) {
	// up操作时找左右节点出现次数最多的颜色， 一样多则加起来
    if (t[lson].num > t[rson].num) {
        t[id].num = t[lson].num;
        t[id].sum = t[lson].sum;
    }
    else if (t[lson].num < t[rson].num) {
        t[id].num = t[rson].num;
        t[id].sum = t[rson].sum;
    }
    else{
        t[id].num = t[lson].num;
        t[id].sum = t[lson].sum + t[rson].sum;
    }
}

void modify(int &id, int l, int r, int p) {
	// 动态开点
    if(!id) id = ++tot;
    if (l == r) {
        t[id].num += 1;
        t[id].sum = l;
        return;
    }
    int mid = l + r >> 1;
    if (p <= mid) modify(lson, l, mid, p);
    else modify(rson, mid + 1, r, p);
    up(id);
}

int merge(int a, int b, int l, int r) {
    if (!a) return b;
    if (!b) return a;
    if (l == r) {
        t[a].sum = l;
        t[a].num += t[b].num;
        // 合并完后子树的信息可以保存在一棵树上，所以直接合并到a上
        return a;
    }
    int mid = l + r >> 1;
    t[a].l = merge(t[a].l, t[b].l, l, mid);
    t[a].r = merge(t[a].r, t[b].r, mid + 1, r);
    up(a);
    return a;
}

void dfs(int u, int fa) {
    for (auto v : G[u]) {
        if (v == fa) continue;
        dfs(v, u);
        // 对于前n个节点，rt[i] = i
        merge(u, v, 1, n);
    }
    // 将自己本身加到权值线段树里
    modify(u, 1, n, cor[u]);
    ans[u] = t[u].sum;
}
void solve() {
    read(n);
    tot = n;
    for (int i = 1; i <= n; ++i) read(cor[i]);
    for (int i = 1, u, v; i < n; ++i) {
        read(u), read(v);
        G[u].push_back(v);
        G[v].push_back(u);
    }
    dfs(1, 0);
    for (int i = 1; i <= n; ++i) {
        printf("%lld%c", ans[i], " \n"[i == n]);
    }
}
int main() {
    int _T = 1;
    while (_T--) solve();
    return 0;
}