HDU 5927 Auxiliary Set（dfs+逆向思维）

Auxiliary Set

这道题本来以为是dfs序加上一些数据结构处理的，但是我看到一个点就没有了这种想法。大家可以发现q次询问每一次就相当于重新给了条件，给出不重要的点，这种如果用刚才的思想处理的话很明显是行不通的，因为刚才的思想需要根据dfs序来构建线段树然后进行求解很明显会超时，我想到了从子节点递推父节点，但是自己从没有做过类似的题目有了这种想法不知道怎么处理于是看了题解，果然是这样的。就是根据结点的深度进行排序，然后再用子节点去推出父节点，然后找出满足第二个条件的点数。

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
const int maxn = 1e5+5;

int head[maxn], dep[maxn], fa[maxn], son[maxn], cnt[maxn], a[maxn], tot;

struct Edge{
    int to, next;
}e[maxn<<1];

void addedge(int from, int to){
    e[tot].to = to;
    e[tot].next = head[from];
    head[from] = tot++;
}

void dfs(int u, int f){
    son[u] = 0;  //孩子数目
    fa[u] = f;   //父亲节点
    for(int i = head[u]; i != -1; i = e[i].next){
        int v = e[i].to;
        if(v == f) continue;
        son[u]++;
        dep[v] = dep[u]+1;  //深度
        dfs(v, u);
    }
}

void init(){
    tot = 0;
    memset(head, -1, sizeof(head));
    memset(dep, 0, sizeof(dep));
    memset(fa, 0, sizeof(fa));
    memset(son, 0, sizeof(son));
}

int cmp(int a, int b){  //按照深度进行排序
    return dep[a] > dep[b];
}

void solve(){
    init();
    int n, m, q, u, v;
    scanf("%d%d", &n, &q);
    for(int i = 1; i < n; i++){
        scanf("%d%d", &u, &v);
        addedge(u, v);
        addedge(v, u);
    }
    dfs(1, 0);
    while(q--){
        scanf("%d", &m);
        for(int i = 0; i < m; i++) scanf("%d", &a[i]), cnt[a[i]] = son[a[i]];
        int ans = n - m;
        sort(a, a+m, cmp);
        for(int i = 0; i < m; i++){
            if(cnt[a[i]] >= 2) ans++; //如果子树中有2个或者以上重要点那么该点也是重要点
            else if(cnt[a[i]] == 0) cnt[fa[a[i]]]--;  //如果该点为根的子树没有重要节点那么父亲节点的子节点中重要节点的个数需要减1
        }
        printf("%d\n", ans);
    }
}

int main(){
    int T;
    scanf("%d", &T);
    for(int kase = 1; kase <= T; kase++){
        printf("Case #%d:\n", kase);
        solve();
    }
    return 0;
}