Hawala Arrests CodeChef - AMR16G

题意 ：一棵树，每个点可以有 0-K 种颜色，每种颜色在树上的点连续的，选择最少的点覆盖K种颜色，

思路 ： 树上贪心，对应到序列上，类似于 k个区间 ，选择最少的点，让每个区间至少包含一个点的经典贪心问题

解决方法是 按照右端点排序，贪心的选择点，那么在树上问题类似，我们把每个颜色块的根 看作区间右端点，按照deep进行排序，贪心的选点即可。

#include<bits/stdc++.h>
using namespace std;
#define N 1000100
#define inf 0x3f3f3f3f
#define go(i,a,b) for(int i=a;i<=b;i++)
vector<int>p[N],col[N];
int T,n,k,u,c,vis[N],deep[N],mi[N],id[N],pos[N],ans;
bool cmp(int a,int b)
{
    return mi[a]>mi[b];
}
void dfs(int u,int dep)
{
    deep[u]=dep;
    for(int i=0; i<p[u].size(); i++)
        dfs(p[u][i],dep+1);
}
int main()
{
    scanf("%d",&T);
    while(T--)
    {
        ans=0;
        scanf("%d %d",&n, &k);
        go(i,1,n)p[i].clear();
        go(i,1,k)
        {
            vis[i]=0,col[i].clear();
            mi[i]=inf,pos[i]=i;
        }
        go(i,2,n)
        {
            scanf("%d", &u);
            p[u].push_back(i);
        }
        dfs(1,1);
        go(i,1,k)
        {
            scanf("%d",&c);
            go(j,1,c)
            {
                scanf("%d",&u);
                col[u].push_back(i);
                if(deep[u]<mi[i])
                {
                    mi[i]=deep[u];
                    id[i]=u;
                }
            }
        }
        sort(pos+1,pos+1+k,cmp);
        go(j,1,k)
        {
            int i=pos[j];
            if(vis[i])continue;
            ans++;
            for(int q=0;q<col[id[i]].size();q++)
            vis[col[id[i]][q]]=1;
        }
        printf("%d\n",ans);
    }
    return 0;
}