【启发式合并】【主席树】HDU - 6504 - Problem E. Split The Tree

题目链接：https://cn.vjudge.net/problem/HDU-6504

做法一：启发式合并，用unorder_map维护每个子树的颜色信息。

需要用到第x个颜色的总数，和第x个颜色在第i棵子树内的个数来判断删除第i棵子树后的答案变化。

#include<bits/stdc++.h>
#define rep(i, a, b) for(int i = (a); i <= (b); i++)
#define pb push_back
using namespace std;
typedef long long ll;
const int N=1e5+100;
vector<int> nxt[N];
int n,ans,val[N],cnt[N],tot,dif[N];     //cnt[i]记录i的总共出现次数 ,dif[i]删除第i棵子树的答案，val[i]记录第i个点的权值
unordered_map<int,int> re[N];       //re[i][x]记录第i棵子树中x的出现次数
void init(int n) {
    memset(cnt,0,sizeof(cnt));
    rep(i, 1, n) {
        nxt[i].clear();
        re[i].clear();
        dif[i] = 0;
    }
    ans = 0;
    tot = 0;
}
int siz[N],son[N];
void dfs(int u,int f) {
    son[u] = 0;                //重儿子
    siz[u] = 1;
    for(auto v:nxt[u]) {
        if(v==f)continue;
        dfs(v,u);
        siz[u] += siz[v];
        if(siz[son[u]]<siz[v]) son[u] = v;   //维护重儿子
    }
}
void dfs2(int u,int f) {
    for(auto v:nxt[u]) {
        if(v==f) continue;
        dfs2(v,u);
    }
    if(siz[u]==1) {
        re[u][val[u]] = 1;
        dif[u] = tot+1;
        if(re[u][val[u]]==cnt[val[u]]) dif[u]--;
        ans = max(ans,dif[u]);
    }
    else {
        swap(re[u],re[son[u]]);
        dif[u] = dif[son[u]];
        for(auto v:nxt[u]) {
            if(v==f||v==son[u]) continue;
            for(auto x:re[v]) {
                if(re[u][x.first]==0) dif[u]++;
                re[u][x.first]+=x.second;
                if(re[u][x.first]==cnt[x.first]) dif[u]--;
            }
        }
        if(re[u][val[u]]==0) dif[u]++;
        re[u][val[u]]++;
        if(re[u][val[u]]==cnt[val[u]]) dif[u]--;
        ans = max(ans,dif[u]);
    }
}
int main() {
   // freopen("a.txt","r",stdin);
    ios::sync_with_stdio(0);
    while(cin>>n) {
        init(n);
        rep(i, 2, n) {
            int f;
            cin>>f;
            nxt[i].pb(f);
            nxt[f].pb(i);
        }
        rep(i, 1, n) {
            cin>>val[i];
            if(cnt[val[i]]==0) tot++;
            cnt[val[i]]++;
        }
        dfs(1,0);
        dfs2(1,0);
        cout<<ans<<endl;
    }
}

做法二：dfs序把树压成一维数组，然后把这个数组复读一份，连在原数组的后面。

这样的话子树就是数组内连续的一部分，删除子树后的剩余状态也是连续的一部分。

然后就是主席树模板题了。