题意 :一棵树,每个点可以有 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;
}