题目大意
给出一颗无根树,每个节点都为白色或黑色。求至少需要将多少白色的结点变成黑色,使得删除完所有白点后,剩余的黑点可构成一棵树。
思路讲解
由于黑点不可变为白点,所以删点后的树中必然包含所有原来为黑色的结点,所以我们可以任取一个黑色的结点作为删点后的树和原来的树的根节点。
因为删点后只能剩下一棵树,所以所有黑点和根节点的路径中包含的所有的点必需为黑色。也就是说如果一个白点在一条黑点与根节点间的路径中,则必需将这个白点变为黑色。
于是我们可以对这颗树进行 dfs
。创建一个返回值类型为 bool
的函数 dfs(x)
。如果 dfs(x)
的返回值为 true
,则节点
x
x
x 所包含的子树中至少有一个黑色的结点,否则返回 false
,即子树中没有黑色的结点。
这样我们遍历时,如果一个结点的直接连接的深度比该点大的点中的 dfs
值为 true
,则说明存在一条黑点与根节点的路径,包含该节点。如果发生这种情况,我们需要将返回值定为 true
,如果该点为白色,则再将这个白点变为黑色,并让答案加
1
1
1。
如果点
x
x
x 为黑色或出现上一段出现的情况,则函数 dfs(x)
返回的值为 true
,否则为 false
。
这里需要注意的是,即使我们确定了返回值,也不能直接返回返回值结束函数。只有遍历完后返回,才能得到正确答案。
算法复杂度为 O ( n ) O(n) O(n)。
代码实现
#include <bits/stdc++.h>
using namespace std;
int n,a[100005],ans;
bool mark[100005];
vector<int> g[100005];
bool dfs(int x){
bool flag=bool(a[x]);
//用flag存储返回值,如果a[x]为黑色,flag为true
mark[x]=true;//标记该结点已被遍历
int len=g[x].size();
for(int i=0;i<len;i++)
if(!mark[g[x][i]])//深度大的点都已被遍历
if(dfs(g[x][i]))//如果出现黑点,返回值赋值为true,如果a[x]为白色,ans就+1,a[x]不管怎样都变成黑色
flag=true,ans+=(a[x]==0),a[x]=1;
return flag;
}
int main(){
cin>>n;
for(int i=1;i<=n;i++)cin>>a[i];
for(int i=1;i<n;i++){
int u,v;
cin>>u>>v;
g[u].push_back(v);
g[v].push_back(u);
}
for(int i=1;i<=n;i++)
if(a[i]){
dfs(i);//如果结点为黑色,则作为根节点遍历
break;
}
cout<<ans;
return 0;
}
完结撒花!
(如果该文章有表述错误或可以改进的地方,欢迎在评论区提出)