#139. 【UER #4】被删除的黑白树
统计很久很久以前,有一棵树加入了 UOJ 群。
这天,在它讨论“一棵树应该怎么旋转”的时候一不小心被删除了,变成了被删除的树。
突然间,它突然发现它失去了颜色,变成了一棵纯白的树。这让它感觉很焦躁,于是它来拜托你给自己染上一些颜色。
我们可以把它描述为一棵 n 个节点的有根树(默认树的根为 1 号节点),所有非根的度数为 1 的节点被称为叶子节点。最开始所有的节点都是白色的。
现在你需要选出一些节点并把这些节点染成黑色的。为了迎合树的审美,你的染色方案必须要满足所有叶子节点到根路径上的黑色节点个数相同。
你发现黑色节点个数越多,树就会越高兴,所以你想要知道在所有合法的染色方案中,黑色节点总个数最多是多少。
输入格式
第一行一个正整数 n 表示树的节点个数。
接下来的 n−1 行,每行是两个整数 u,v (1≤u,v≤n,u≠v) 表示树上的一条边。
输出格式
一个整数,表示在所有合法方案中黑色节点的最多数量。
样例一
input
7 1 2 1 3 2 4 2 5 3 6 3 7
output
7
explanation
所有节点都染黑是一个合法的方案。
样例二
input
6 1 2 2 3 1 4 4 5 5 6
output
5
explanation
一个最优的染色方案是把集合 {1,2,3,4,5} 染黑。
这题的思路比较好想,但是操作不好想(简直骚的一逼)
1:首先找出最浅的叶子节点
2:为了使黑色节点尽量大 所以对于当前节点尽量使底部节点被染色 但又不得不考虑 最浅的叶子节点对其的需要
所以 设置一个变量记录还需要多少个染色 如果当前的最浅深度与变量相同即不得不染色 即产生贡献
#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<vector>
#include<map>
#include <bits/stdc++.h>
using namespace std;
const int N = 300000+10;
typedef long long LL;
const LL mod = 998244353;
vector<int>p[N];
int d[N], b[N];
void dfs1(int u,int f)
{
b[u]=N;
for(int i=0;i<p[u].size();i++)
{
int v=p[u][i];
if(v==f) continue;
dfs1(v,u);
b[u]=min(b[u],b[v]+1);
}
if(b[u]==N) b[u]=1;
return ;
}
void dfs2(int u,int f,int k)
{
d[u]=0;
for(int i=0;i<p[u].size();i++)
{
int v=p[u][i];
if(v==f) continue;
dfs2(v,u,min(b[u]-1,k));
d[u]+=d[v];
}
if(b[u]==k) d[u]++;
return ;
}
int main()
{
int n;
scanf("%d", &n);
for(int i=1;i<n;i++)
{
int x, y;
scanf("%d %d",&x, &y);
p[x].push_back(y),p[y].push_back(x);
}
dfs1(1,-1);
dfs2(1,-1,b[1]);
cout<<d[1]<<endl;
return 0;
}