bzoj5189 [Usaco2018 Jan]Cow at Large（树+贪心）

给定一棵有根树，wrx一开始在根，
每个叶子节点都是出口，你可以在每个出口
放一个守卫，每一单位时间内，wrx和守卫
都可以移动到相邻的一个点，如果某一时刻
守卫与wrx相遇了（在边上或点上均算），则
wrx将被抓住。问你为了保证抓住wrx，
最少需要几个守卫。
显然最多叶子结点个数个，每个守卫每次
都往他的父亲走是最优的。对于每个点i
我们预处理出mn[i],表示i的子树中距i最近的
叶子节点距i的距离。显然如果mn[i]<=dep[i]，
则i的子树均可以用这一个守卫守住，否则要
去下面用更多的守卫。算一下答案即可。

#include 
#include 
#include 
using namespace std;
#define ll long long
#define inf 0x3f3f3f3f
#define N 100010
inline int read(){
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();
    return x*f;
}
int n,rt,h[N],num=0,fa[N],dep[N],mn[N],ans=0;
struct edge{
    int to,next;
}data[N<<1];
inline void dfs(int x){
    int son=0;
    for(int i=h[x];i;i=data[i].next){
        int y=data[i].to;if(y==fa[x]) continue;son++;
        fa[y]=x;dep[y]=dep[x]+1;dfs(y);mn[x]=min(mn[x],mn[y]+1);
    }if(!son) mn[x]=0;
}
inline void calc(int x){
    if(mn[x]<=dep[x]){ans++;return;}
    for(int i=h[x];i;i=data[i].next){
        int y=data[i].to;if(y==fa[x]) continue;calc(y);
    }
}
int main(){
    n=read();rt=read();memset(mn,inf,sizeof(mn));
    for(int i=1;iint x=read(),y=read();
        data[++num].to=y;data[num].next=h[x];h[x]=num;
        data[++num].to=x;data[num].next=h[y];h[y]=num;
    }dfs(rt);calc(rt);
    printf("%d\n",ans);
    return 0;
}