Interesting Vertices

Interesting Vertices（前向星+思维+dfs回溯)

参考博客：https://blog.csdn.net/I_believe_CWJ/article/details/102472012

题目大意：给你一课有n个节点的树，其中有k个节点被染色，求有多少个节点满足自身没有被染色
并且它的每棵子树中都至少有一个节点被染色。
解题思路：dfs回溯类似求树的重心的方式求解，dfsdfs回溯可以得到每个节点的它的子树中是否都有染色的点
并统计所有子树中染色点的个数sum，然后走向父亲的那棵子树中被染色的点数就为k−sum，
这样即可判断此节点是否满足条件。

AC_Code

 #include <iostream>
 #include <cstdio>
 #include <cstring>
 #include <ctime>
 #include <cmath>
 #include <vector>
 #include <map>
 #include <vector>
 #include <algorithm>
 #define bug printf("*********\n");
 #define mem0(a) memset(a, 0, sizeof(a));
 #define mem1(a) memset(a, -1, sizeof(a));
 #define ios ios::sync_with_stdio(false);
 #define pb(x) push_back(x)
 using namespace std;
 typedef long long LL;
 const LL mod = 1e9 + ;
 const int inf = 1e9 + ;
 const LL INF = 1e18 + ;
 const double eps = acos(-1.0);

 int n, k, cnt;
 int c[];///标记是否是染色点
 int head[];///前向星使用
 int vis[];///标记是否访问过
 int sz[];///sz[i]表示第i个节点的子树中一共有多少个被染色

 struct edge {
     int to, nxt;
 }e[*];

 void add(int u, int v) {
     e[cnt].to = v;
     e[cnt].nxt = head[u];
     head[u] = cnt ++;
 }

 vector<int> ans;

 void dfs(int u) {
     vis[u] = ;
     sz[u] = c[u];
     int flag = ;
     for(int i = head[u]; ~i; i = e[i].nxt) {
         int en = e[i].to;
         if(vis[en]) continue;
         dfs(en);
         if(!sz[en]) flag = ;///此子树没有被染色
         sz[u] += sz[en];
     }
     ///k-sz[u]>0 是指u的父亲节点所带领的那棵子树有被染色
     ///！c[u]是指满足改点没有被染色
     ///flag表示点的所有子树都有被染色
     ///u==1是因为1节点没有父亲节点了
     if(!c[u] && flag && (k-sz[u]> || u == )) ans.pb(u);
 }

 int main() {
     int x, y;
     while(~scanf("%d%d", &n, &k)) {
         mem0(vis);mem0(c);mem1(head);
         cnt = ;
         ans.clear();
         for(int i = ; i < k; i ++) {
             scanf("%d", &x);
             c[x] = ;
         }
         for(int i = ; i < n-; i ++) {
             scanf("%d%d", &x, &y);
             add(x, y), add(y, x);
         }
         dfs();///假装以1为树的总根
         sort(ans.begin(), ans.end());
         printf("%d\n", ans.size());
         for(auto i : ans) {
             printf("%d ", i);
         }
         printf("\n");
     }
     return ;
 }