Description
一棵树,可以进行染色,被染色的点可以控制与它距离不超过 \(k\) 的所有点,问控制整棵树最少需要染几个点.
Sol
贪心.
记录一下最深的未染色点和最浅的染色点,判断一下能否在子树中就完成,不能的话就把权值赋成最深未染色点深度+1,能的话就赋成染色点深度+1.
需要特判一下根.
Code
#include<cstdio>
#include<vector>
#include<ctime>
#include<cstdlib>
#include<iostream>
using namespace std;
const int N = 100005;
int n,k,ans;
int f[N];
vector<int> g[N];
inline int in(int x=0,char ch=getchar()){ while(ch>'9' || ch<'0') ch=getchar();
while(ch>='0' && ch<='9') x=x*10+ch-'0',ch=getchar();return x; }
void DFS(int u,int fa){
int mx=0,mi=0;
for(int i=0,v;i<g[u].size();i++) if((v=g[u][i])!=fa){
DFS(v,u);
mx=max(mx,f[v]),mi=min(mi,f[v]);
}
if(mx+1 > k) f[u]=-k,ans++;
else if(mx+mi+1>0) f[u]=mx+1;
else f[u]=mi+1;
}
int main(){
n=in(),k=in();
for(int i=1,u,v;i<n;i++) u=in()+1,v=in()+1,g[u].push_back(v),g[v].push_back(u);
DFS(1,-1);
if(f[1]>0) ans++;
cout<<ans<<endl;
return 0;
}