首页 > 代码库 > 51Nod 1378 夹克老爷的愤怒

51Nod 1378 夹克老爷的愤怒

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;}

  

51Nod 1378 夹克老爷的愤怒