HDU - 5927 F - Auxiliary Set (DFS判LCA)

原题地址:http://acm.hdu.edu.cn/showproblem.php?pid=5927

题意:
一棵节点数为n的有根数，根节点为1，一开始所有的点都是重点，接下来有q次询问，每次询问把m个点变为轻点，问你树中还有多少个重点。
重点应该满足的条件为：
1.它本身是重点。
2.它为两个重点的最近公共祖先。
每次询问之后在下次询问前，所有的点都恢复为重点。

思路: 首先我们可以知道,对于一个非重要点 只要其至少两颗子树上分别有两个重要点, 那这个非重要点一定是他们的LCA,那么结果数量就会+1.所以依靠这个特点,我们对可以对每次询问的非重要节点按照深度排序,然后一次判断它子树的重要节点数,如果>=2,那么结果+1,如果==1那么没有影响,如果==0,那么就必须要将这个节点的父亲节点的儿子数-1,因为你当前节点没有重要节点,对答案没有任何的贡献.

#include <bits/stdc++.h>
#define eps 1e-8
#define INF 0x3f3f3f3f
#define PI acos(-1)
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
#define CLR(x,y) memset((x),y,sizeof(x))
#define fuck(x) cerr << #x << "=" << x << endl

using namespace std;
typedef long long ll;
typedef unsigned long long ull;
const int seed = 131;
const int maxn = 1e5 + 5;
const int mod = 998244353;
int t;
struct node {
    int v, nxt;
} e[maxn << 1];

int tot, n, q, head[maxn];

void add_edge(int u, int v) {
    e[tot].v = v;
    e[tot].nxt = head[u];
    head[u] = tot++;
}
int dep[maxn], son[maxn], tmp[maxn]; //dep是深度,son是儿子节点数量,tmp复制son数组
int f[maxn];//f是父亲数组
void dfs(int u, int fa, int d) {
    dep[u] = d;
    f[u] = fa;
    for (int i = head[u]; ~i; i = e[i].nxt) {
        int v = e[i].v;
        if (v == fa)
            continue;
        son[u]++;
        dfs(v, u, d + 1);
    }
}
bool cmp(int a, int b) {
    return dep[a] > dep[b];
}
void init_head() {
    tot = 0;
    for (int i = 1; i <= n; i++)
        head[i] = -1;
    for (int i = 1; i <= n; i++)
        son[i] = 0;
}
int uu[maxn];//不重要集合
int main() {
    scanf("%d", &t);
    for (int cas = 1; cas <= t; ++cas) {
        printf("Case #%d:\n", cas);

        scanf("%d%d", &n, &q);
        init_head();
        for (int i = 2; i <= n; ++i) {
            int u, v;
            scanf("%d%d", &u, &v);
            add_edge(u, v);
            add_edge(v, u); //要建双向边
        }
        dfs(1, 1, 1);
        while (q--) {
            int num;
            scanf("%d", &num);
            int ans = n - num;
            for (int i = 1; i <= num; ++i) {
                scanf("%d", &uu[i]);
                tmp[uu[i]] = son[uu[i]];
            }
            sort(uu + 1, uu + 1 + num, cmp);
            for (int i = 1; i <= num; ++i) {
                if (tmp[uu[i]] >= 2) {
                    ans++;
                } else if (tmp[uu[i]] == 1) {
                    continue;
                } else {
                    tmp[f[uu[i]]]--;
                }
            }
            printf("%d\n", ans);
        }
    }
    return 0;
}