HDU 5927 Auxiliary Set (dfs)

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

题意：

        给你一棵树，其中有一些’不重要‘的点，要是这些’不重要‘的点的子树中有两个重要的点的LCA是这个点，那么这个点就是重要的点。每次询问 问你重要的点有多少？

思路：

        用不重要的点建图，要是这个不重要点上的相邻子树至少有两个有重要的点不在同一子树上，那么这个不重要的点才变成重要的点。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <vector>
#include <map>
using namespace std;
const int N = 1e5 + 5;
vector <int> G[N];
int d[N], a[N], par[N], d2[N], cnt, son[N];
bool vis[N];
//son[u]表示u节点相邻的子树个数
//原图：d[u]表示以u节点为root的子树节点个数
//不重要点建的图：d2[u]表示以u节点为root的子树节点个数
//cnt表示不重要的点的个数

void init(int n) {
    for(int i = 1; i <= n; ++i) {
        G[i].clear();
        vis[i] = false;
    }
}

void dfs1(int u, int p) {
    d[u] = 1;
    par[u] = p;
    son[u] = 0;
    for(int i = 0; i < G[u].size(); ++i) {
        int v = G[u][i];
        if(v == p) 
            continue;
        dfs1(v, u);
        d[u] += d[v];
        ++son[u];
    }
}

void dfs2(int u) {
    d2[u] = 1;
    vis[u] = true;
    int num = 0, num2 = 0;
    for(int i = 0; i < G[u].size(); ++i) {
        int v = G[u][i];
        if(!vis[v])
            dfs2(v);
        d2[u] += d2[v];
        if(d[v] > d2[v]) {
            ++num;
        }
        num2++;
    }
    if(num + (son[u] - num2) > 1) //符合条件 (不同子树中重要点)>=2;
        return ;
    else
        cnt++;
}

int main()
{
    int t, n, m, u, v, q;
    scanf("%d", &t);
    for(int ca = 1; ca <= t; ++ca) {
        scanf("%d %d", &n, &m);
        init(n);
        for(int i = 1; i < n; ++i) {
            scanf("%d %d", &u, &v);
            G[u].push_back(v);
            G[v].push_back(u);
        }
        printf("Case #%d:
", ca);
        dfs1(1, 0);
        init(n);
        while(m--) {
            scanf("%d", &q);
            for(int i = 1; i <= q; ++i) {
                scanf("%d", a + i);
                vis[a[i]] = true;
            }
            for(int i = 1; i <= q; ++i) {
                if(vis[par[a[i]]]) {
                    G[par[a[i]]].push_back(a[i]);
                }
            }
            for(int i = 1; i <= q; ++i) {
                vis[a[i]] = false;
            }
            cnt = 0;
            for(int i = 1; i <= q; ++i) {
                if(!vis[a[i]]) {
                    dfs2(a[i]);
                }
            }
            for(int i = 1; i <= q; ++i) {
                vis[a[i]] = false;
                G[a[i]].clear();
            }
            printf("%d
", n - cnt);
        }
    }
    return 0;
}