B - Interesting Vertices ( DFS应用 )
题目大意:给你一课有n个节点的树,其中有k个节点被染色,求有多少个节点满足自身没有被染色并且它的每棵子树中都至少有一个节点被染色。
解题思路:dfs
dfs回溯类似求树的重心的方式求解,dfs回溯可以得到每个节点的它的子树中是否都有染色的点并统计所有子树中染色点的个数sum,然后走向父亲的那棵子树中被染色的点数就为k−sum,这样即可判断此节点是否满足条件。
代码:
#include <bits/stdc++.h>
using namespace std;
const int maxn = 6e5+10;
int via[maxn];
int head[maxn];
int cnt,n,k;
set<int> ans;
struct node {
int to,nxt;
}e[maxn];
void addage( int u, int v )
{
e[cnt].to = v;
e[cnt].nxt = head[u];
head[u] = cnt++;
e[cnt].to = u;
e[cnt].nxt = head[v];
head[v] = cnt++;
}
int dfs( int v0, int father )
{
int sum = 0;
if ( via[v0]==1 ) sum ++;
int isp = 1;
for ( int i=head[v0]; i!=-1; i=e[i].nxt ) {
int x = e[i].to;
if ( x==father ) continue ;
int p = dfs(x,v0);
if ( p==0 ) {
isp = 0;
}
sum += p;
}
if ( isp==1 && via[v0]==0 && sum!=k ) { // sum!=k的意思是,v0可能存在父亲分支,但上面的for循环没有走父亲分支
ans.insert(v0); // 所以如果sum==k, 就代表父亲分支一定没有染色的点了,
}
else if ( isp==1&&v0==1&&via[v0]==0 ) { // 没有父亲分支
ans.insert(v0);
}
return sum;
}
int main()
{
int i,j,x;
memset(via,0,sizeof(via));
memset(head,-1,sizeof(head));
cnt = 0;
cin >> n >> k;
ans.clear();
for ( i=0; i<k; i++ ) {
scanf("%d",&x);
via[x] = 1;
}
for ( i=1; i<n; i++ ) {
int u,v;
scanf("%d %d",&u,&v);
addage(u,v);
}
dfs(1,-1);
cout << ans.size() << endl;
for ( set<int>::iterator i=ans.begin(); i!=ans.end(); i++ ) {
cout << *i << " ";
}
return 0;
}
/*
5 4
5 4 1 2
3 1
3 2
3 4
3 5
*/