hdu 5927 Auxiliary Set

Auxiliary Set

题目链接：

http://acm.split.hdu.edu.cn/showproblem.php?pid=5927

解题思路：

题目大意：

给你一个根节点为1的树，定义一个节点为重要的节点至少满足两个条件中的一个条件：
1：自己本身是重要的点
2：两个以上的子孙是重要的点
接着给你q组询问，给出m个不重要的点，让你输出树上有多少重要的点？

算法思想：

对unimportant node按逆DFS序排序.
接着遍历此集合, 同时对每个节点u进行维护,维护u满足如下条件的儿子v的数目:
子树v中不全是un-node。
不妨将此数目记作cnt[u]，维护的方法是：
对某个u, 若 cnt[u]==0 就 --cnt[pa[u]]。
最后, 满足 cnt[u]≥2 的u的数目即为所求。

AC代码：

#include <bits/stdc++.h>using namespace std;const int N = 100005;int tail;int dep[N],pa[N];int cnt[N],_cnt[N];vector<int> g[N],un;//un is unimportant nodesvoid dfs(int u,int fa){    dep[u] = ++tail;    pa[u] = fa;    for(auto v:g[u]){        if(v != fa){            ++cnt[u];            dfs(v,u);        }    }}bool cmp(int x,int y){    return dep[x] > dep[y];}int main(){    int t = 0,T;    scanf("%d",&T);    while(T--){        int n,q;        scanf("%d%d",&n,&q);        for(int i = 0; i <= n; ++i){            g[i].clear();            cnt[i] = 0;        }        int u,v;        for(int i = 1; i < n; ++i){            scanf("%d%d",&u,&v);            g[u].push_back(v);            g[v].push_back(u);        }        tail = 0;        dfs(1,0);        printf("Case #%d:\n", ++t);        while(q--){            int x,m;            scanf("%d",&m);            un.clear();            while(m--){                scanf("%d",&x);                un.push_back(x);            }            sort(un.begin(),un.end(),cmp);            for(auto x:un){                _cnt[x] = cnt[x];            }            for(auto x:un){                if(_cnt[x] == 0){                    --_cnt[pa[x]];                }            }            int ans = 0;            for(auto x:un){                ans += _cnt[x] >= 2;            }            ans += n-un.size();            printf("%d\n",ans);        }    }    return 0;}