CSU 1811 Tree Intersection(启发式合并)

$今天学了好多新知识呀喵喵喵$
$启发式合并，其实只是一种思想，就是在一些交换的时候，按照大小的顺序，把小的往大的上加$
$这样的复杂度是对于合并出来n个东西，是O(nlogn)的$
$为什么呢，最坏情况下是操作了$
$\frac{n}{2}+2*\frac{n}{4}+4*\frac{n}{8}+\dots\le\frac{n}{2}+\frac{n}{3}+\frac{n}{4}+\dots=nlogn$
$然后对于一棵树，如果每个子节点都有一个信息，并且信息的种类很多，不好记录$
$那么可以直接用map对每个子树的信息进行记录，然后回溯上去的时候$
$父节点加上子节点信息的时候，把小的往大的加$
$然后每次新出来一个颜色，就+1，如果颜色满了，就-1$
$每次新搜一个节点的时候，如果子节点更大，那么ans也要赋值成子节点的ans$
$想一下就知道了，知道启发式合并之后，难点就是ans的变化了$

---

代码：

#include <map>
#include <set>
#include <stack>
#include <queue>
#include <cmath>
#include <string>
#include <vector>
#include <cstdio>
#include <cctype>
#include <cstring>
#include <sstream>
#include <cstdlib>
#include <iostream>
#include <algorithm>
#pragma comment(linker,"/STACK:102400000,102400000")

using namespace std;
#define   MAX           100005
#define   MAXN          1000005
#define   maxnode       205
#define   sigma_size    2
#define   lson          l,m,rt<<1
#define   rson          m+1,r,rt<<1|1
#define   lrt           rt<<1
#define   rrt           rt<<1|1
#define   middle        int m=(r+l)>>1
#define   LL            long long
#define   ull           unsigned long long
#define   mem(x,v)      memset(x,v,sizeof(x))
#define   lowbit(x)     (x&-x)
#define   pii           pair<int,int>
#define   bits(a)       __builtin_popcount(a)
#define   mk            make_pair
#define   limit         10000

//const int    prime = 999983;
const int    INF   = 0x3f3f3f3f;
const LL     INFF  = 0x3f3f;
//const double pi    = acos(-1.0);
const double inf   = 1e18;
const double eps   = 1e-9;
const LL     mod   = 1e9+7;
const ull    mx    = 133333331;

/*****************************************************/
inline void RI(int &x) {
      char c;
      while((c=getchar())<'0' || c>'9');
      x=c-'0';
      while((c=getchar())>='0' && c<='9') x=(x<<3)+(x<<1)+c-'0';
 }
/*****************************************************/

struct Edge{
    int v,next,id;
}edge[MAX*2];
int head[MAX];
int ans[MAX];
int sum[MAX];
int c[MAX];
map<int,int> cnt[MAX];
int tot;

void init(){
    mem(head,-1);
    mem(ans,0);
    mem(sum,0);
    tot=0;
}

void add_edge(int a,int b,int id){
    edge[tot]=(Edge){b,head[a],id};
    head[a]=tot++;
}

void dfs(int u,int fa,int id){
    cnt[u][c[u]]=1;
    if(sum[c[u]]>1) ans[id]++;
    for(int i=head[u];i!=-1;i=edge[i].next){
        int v=edge[i].v;
        if(v==fa) continue;
        dfs(v,u,edge[i].id);
        if(cnt[u].size()<cnt[v].size()){
            swap(cnt[u],cnt[v]);
            ans[id]=ans[edge[i].id];
        }
        for(map<int,int>::iterator it=cnt[v].begin();it!=cnt[v].end();it++){
            int x=it->first,y=it->second;
            if(cnt[u].count(x)){
                cnt[u][x]+=y;
                if(cnt[u][x]==sum[x]) ans[id]--;
            }
            else{
                cnt[u][x]=y;
                if(cnt[u][x]<sum[x]) ans[id]++;
            }
        }
    }
}

int main(){
    //freopen("in.txt","r",stdin);
    int n;
    while(cin>>n){
        init();
        for(int i=1;i<=n;i++){
            scanf("%d",&c[i]);
            cnt[i].clear();
            sum[c[i]]++;
        }
        //cout<<all<<endl;
        for(int i=1;i<n;i++){
            int a,b;
            scanf("%d%d",&a,&b);
            add_edge(a,b,i);
            add_edge(b,a,i);
        }
        dfs(1,-1,0);
        for(int i=1;i<n;i++) printf("%d\n",ans[i]);
    }
    return 0;
}