HDU 5927 Auxiliary Set (树形dp)

题意：判断每个给出节点是否为剩余节点的lca
朴素的想法就是对每个点进行dfs判断各个子树上是否有重要节点。
然而重复访问节点的次数过多，显然超时。
于是想到对每个节点记录存在重要节点的子树个数，这样每次只要检查当前节点是否符合即可。
同时对每次询问给出的节点，可以自底向上修改cnt值。

注意：只有当前节点为不重要节点且cnt = 0时才修改其父亲的值。

#include <bits/stdc++.h>

using namespace std;
const int maxn = 100010;

struct Edge {
    int to, next;
}edges[maxn<<1];
int head[maxn], tot;
int fa[maxn], son[maxn], d[maxn], cnt[maxn], a[maxn];

void AddEdge(int u, int v) {
    edges[tot].to = v;
    edges[tot].next = head[u];
    head[u] = tot++;
}

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

void dfs(int u, int f) {
    son[u] = 0;
    fa[u] = f;
    for(int i = head[u]; i != -1; i = edges[i].next) {
        int v = edges[i].to;
        if(v == f) continue;
        son[u]++;
        d[v] = d[u]+1;
        dfs(v, u);
    }
}

bool cmp(int a, int b) { return d[a] > d[b]; }

int main() {
    int t, kase = 1;
    scanf("%d", &t);
    while(t--) {
        init();
        printf("Case #%d:\n", kase++);
        int n, q;
        scanf("%d%d", &n, &q);
        int u, v;
        for(int i = 1; i < n; ++i) {
            scanf("%d%d", &u, &v);
            AddEdge(u, v);
            AddEdge(v, u);
        }
        dfs(1, 0);
        int m;
        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++;
                else if(cnt[a[i]] == 0) cnt[fa[a[i]]]--;
            }
            printf("%d\n", ans);
        }
    }
    return 0;
}