Codeforces 600 E. Lomsat gelral (dfs启发式合并map)

题目链接：http://codeforces.com/contest/600/problem/E

给你一棵树，告诉你每个节点的颜色，问你以每个节点为根的子树中出现颜色次数最多的颜色编号和是多少。

最容易想到的是n*n的暴力，但是会超时。所以这里用到类似并查集的合并，对于每个子树颜色种数少的合并到颜色种数大的当中。

不懂的看代码应该可以明白。

 //#pragma comment(linker, "/STACK:102400000, 102400000")
 #include <algorithm>
 #include <iostream>
 #include <cstdlib>
 #include <cstring>
 #include <cstdio>
 #include <vector>
 #include <cmath>
 #include <ctime>
 #include <list>
 #include <set>
 #include <map>
 using namespace std;
 typedef long long LL;
 typedef pair <int, int> P;
 const int N = 1e5 + ;
 map <int, int> cnt[N]; //first代表颜色编号，second代表出现次数
 map <int, LL> sum[N]; //first代表出现次数，second代表颜色编号之和
 LL ans[N];
 int head[N], tot, a[N];
 struct Edge {
     int next, to;
 }edge[N << ];

 void init() {
     memset(head, -, sizeof(head));
 }

 inline void add(int u, int v) {
     edge[tot].next = head[u];
     edge[tot].to = v;
     head[u] = tot++;
 }

 void dfs(int u, int p) {
     cnt[u][a[u]] = ; //初始化：a[u]颜色出现一次
     sum[u][] = (LL)a[u]; //初始化：出现一次的颜色之和为a[u]
     for(int i = head[u]; ~i; i = edge[i].next) {
         int v = edge[i].to;
         if(v == p)
             continue;
         dfs(v, u);
         if(cnt[u].size() < cnt[v].size()) { //颜色种类少的合并到颜色种类多的，u为颜色种类多的子树
             swap(cnt[u], cnt[v]);
             swap(sum[u], sum[v]);
         }
         for(auto it: cnt[v]) {
             cnt[u][it.first] += it.second; //it.first颜色出现次数合并累加
             int temp = cnt[u][it.first]; //temp为it.first颜色次数
             sum[u][temp] += (LL)it.first; //累加出现temp次的颜色
         }
         cnt[v].clear(); //清空
         sum[v].clear();
     }
     ans[u] = sum[u].rbegin()->second; //最大出现次数的颜色之和
 }

 int main()
 {
     init();
     int n, u, v;
     scanf("%d", &n);
     for(int i = ; i <= n; ++i) {
         scanf("%d", a + i);
     }
     for(int i = ; i < n; ++i) {
         scanf("%d %d", &u, &v);
         add(u, v);
         add(v, u);
     }
     dfs(, -);
     for(int i = ; i <= n; ++i) {
         printf("%lld%c", ans[i], i == n? '\n': ' ');
     }
     return ;
 }