题意:在一颗1为根的树上,有很少一部分的不重要的点,其他都是重要的点,问满足条件的点有多少个?
满足条件有两种情形:1.为重要的点。2.他的亲孩子至少有两个不同的重要的点。
思路:先dfs序预处理每个点儿子个数,和父亲是谁。在对于比较少的m个不重要点,求是否满足有两个不同的重要点孩子。这个求法是O(m)。
怎么做呢?对m个不重要点深度排序,从最深的点开始,满足孩子数>=2就答案+1,如果这个点的重要儿子数为0,那么通过预处理出找到它的父亲节点,并把他父亲节点的儿子数组减一。因为这样维护后,每个不重要点的儿子个数就是都是重要点个数。
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
const int maxn=100000+10;
int f[maxn],n,q,a[maxn];
template<int N,int M>//N点的个数,M边的个数
struct Graph
{
int top;
struct Vertex{
int head;
}V[N];
struct Edge{
int v,next;
}E[M];
void init(){
memset(V,-1,sizeof(V));
top = 0;
}
void add_edge(int u,int v){
E[top].v = v;
E[top].next = V[u].head;
V[u].head = top++;
}
};
Graph<100000+10,100000*2+10> g;
int num[maxn],num1[maxn],level[maxn];
void dfs(int cur,int father){
level[cur]=1;
for(int i=g.V[cur].head;i!=-1;i=g.E[i].next){
int son=g.E[i].v;
if(son==father) continue;
dfs(son,cur);
level[cur]=max(level[son]+1,level[cur]);
num[cur]+=1;
f[son]=cur;
}
}
bool cmp(int x,int y){
return level[x]<level[y];
}
int main(){
int T;scanf("%d",&T);
int cas=1;
while(T--){
scanf("%d%d",&n,&q);
g.init();
memset(num,0,sizeof(num));
for(int i=0;i<n-1;i++){
int a,b;scanf("%d%d",&a,&b);
g.add_edge(a,b);
g.add_edge(b,a);
}
dfs(1,0);
printf("Case #%d:\n",cas++);
for(int i=0;i<q;i++){
int m;scanf("%d",&m);
int ans=n-m;
for(int j=1;j<=m;j++) scanf("%d",&a[j]);
for(int j=1;j<=m;j++) num1[a[j]]=num[a[j]];
sort(a+1,a+m+1,cmp);
for(int j=1;j<=m;j++){
if(num1[a[j]]>=2) ans++;
if(num1[a[j]]==0) num1[f[a[j]]]--;
}
printf("%d\n",ans);
}
}
}