启发式合并 && U41492 树上数颜色

传送门

---

启发式合并

通俗点讲，就是将两个数据合并时，小的合并到大的里，这样就可以节约时间。

解题思路

再看看这个题，就是个板子题。

在两个节点信息合并起来时，让颜色数少的节点合并到颜色点多的节点上，实际操作时则是都是儿子的信息合并到父亲上，但是若儿子信息多于父亲，则交换一下信息（swap可做到O(1)）。

可以用map来操作，节约空间。　

AC代码

#include<iostream>
#include<algorithm>
#include<cmath>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<queue>
#include<set>
#include<map>
#include<vector>
#include<iomanip>
#include<ctime>
#include<stack>
using namespace std;
const int maxn=100005;
int n,m,cnt,p[maxn],a[maxn],ans[maxn];
map<int,int> mp[maxn]; 
struct node{
    int v,next;
}e[maxn*2];
void insert(int u,int v){
    cnt++;
    e[cnt].v=v;
    e[cnt].next=p[u];
    p[u]=cnt;
}
void dfs(int u,int fa){
    mp[u][a[u]]=1;
    ans[u]=1;
    for(int i=p[u];i!=-1;i=e[i].next){
        int v=e[i].v;
        if(v==fa) continue;
        dfs(v,u);
        if(ans[v]>ans[u]){//启发式合并
            ans[u]=ans[v];
            swap(mp[v],mp[u]);
        }
        for(map<int,int>::iterator it=mp[v].begin();it!=mp[v].end();it++){
            if(mp[u][it->first]) continue;
            else mp[u][it->first]=1,ans[u]++;
        }
    }
}
int main()
{
    memset(p,-1,sizeof(p));
    cin>>n;
    for(int i=1;i<n;i++){
        int u,v;
        scanf("%d%d",&u,&v);
        insert(u,v);
        insert(v,u);
    }
    for(int i=1;i<=n;i++){
        cin>>a[i];
    }
    dfs(1,-1);
    cin>>m;
    for(int i=1;i<=m;i++){
        int x;
        cin>>x;
        cout<<ans[x]<<endl;
    }
    return 0;
}