HDU 5927 Auxiliary Set【DFS】（2016CCPC东北赛F题）

题目链接：http://acm.split.hdu.edu.cn/showproblem.php?pid=5927

【中文题意】一棵节点数为n的有根数，根节点为1，一开始所有的点都是重点，接下来有q次询问，每次询问把m个点变为轻点，问你树中还有多少个重点。
重点应该满足的条件为：
1.它本身是重点。
2.它为两个重点的最近公共祖先。
每次询问之后在下次询问前，所有的点都恢复为重点。
【思路分析】我们可以先从根节点对整棵树进行深搜一次，预处理出每个节点的儿子节点的个数，以及每个节点的父节点是哪个节点和这个节点所处的深度。然后我们每次查询的时候，按照深度从大到小的顺序对m个点进行排序，然后依次对这m个点每个节点进行处理，首先判断这个点是不是有两个儿子，假如超过两个的话，那么这个节点为重点，如果这个节点没有子节点，那么它的父节点的儿子的数目应当减一，然后完成这m个点的处理。
【AC代码】

#include<cstdio>
#include<cstring>
#include<cmath>
#include<queue>
#include<vector>
#include<algorithm>
using namespace std;
vector<int >G[100005];
int n,m,q;
int book[100005],son[100005],fa[100005],a[100005],de[100005],son2[100005];


void dfs(int u,int depth,int pre)//预处理
{
    de[u]=depth;
    fa[u]=pre;
    son[pre]++;
    son2[pre]=son[pre];
    book[u]=1;
    for(int i=0; i<G[u].size(); i++)
    {
        int v=G[u][i];
        if(book[v]==0)
        {
            dfs(v,depth+1,u);
        }
    }
}

bool cmp(int x,int y)
{
    return de[x]>de[y];
}

int main()
{
    int t,iCase=0;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d %d",&n,&m);
        int u,v;
        //G.resize(n+1);
        memset(book,0,sizeof(book));
        memset(son,0,sizeof(son));
        memset(fa,0,sizeof(fa));
        for(int i=1; i<=n-1; i++)
        {
            scanf("%d%d",&u,&v);
            G[u].push_back(v);
            G[v].push_back(u);
        }
        dfs(1,1,0);/*for(int i=1;i<=n;i++)
        {
            printf("%d ",son[i]);
        }*/
        //printf("\n----------------------\n");
        printf("Case #%d:\n",++iCase);
        for(int i=1; i<=m; i++)
        {
            scanf("%d",&q);
            for(int j=0; j<q; j++)
            {
                scanf("%d",&a[j]);
            }
            sort(a,a+q,cmp);
            int ans=0;
            for(int j=0; j<q; j++)
            {
                // printf("+++%d\n",a[j]);
                if(son[a[j]]>=2)
                {
                    // printf("***1\n");
                    ans++;
                }
                else if(son[a[j]]==0)
                {
                    //   printf("***2\n");
                    son[fa[a[j]]]--;
                }
            }
            printf("%d\n",n-q+ans);
            for(int j=0; j<q; j++)//回复初始状态
            {
                son[fa[a[j]]]=son2[fa[a[j]]];
            }
        }
        for(int j=1; j<=n; j++)
        {
            G[j].clear();
        }
    }
    return 0;
}