省选专练[USACO18JAN]Cow at Large Gold

简单的树形DP

一次DFS求出深度和当前距离叶子的最近距离

第二次DFS求出所有深度小于距离最近叶子的点

#include<bits/stdc++.h>
using namespace std;
const int N=1e5+10;
struct Front_star{
    int u,v,w,nxt;
}e[N*4];
int cnt=0;
int first[N]={};
void add(int u,int v){
    cnt++;
    e[cnt].u=u;
    e[cnt].v=v;
    e[cnt].nxt=first[u];
    first[u]=cnt;
}
int mn[N]={};
int dep[N]={};
int n,k;
void dfs1(int u,int fa){
    for(int i=first[u];i;i=e[i].nxt){
        int v=e[i].v;
        if(v==fa)continue;
        dep[v]=dep[u]+1;
        dfs1(v,u);
        if(mn[u]==0)mn[u]=mn[v]+1;
        mn[u]=min(mn[u],mn[v]+1);
    }
}
int ans=0;
void dfs2(int u,int fa){
//	cout<<dep[u]<<" "<<mn[u]<<'\n';
    if(dep[u]>=mn[u]){
        ans++;
        return;
    }
    for(int i=first[u];i;i=e[i].nxt){
        int v=e[i].v;
        if(fa==v)continue;
        dfs2(v,u);
    }
}
int main(){
    scanf("%d%d",&n,&k);
    for(int i=1;i<n;i++){
        int u,v;
        scanf("%d%d",&u,&v);
        add(u,v);
        add(v,u);
    }
    dfs1(k,0);
    dfs2(k,0);
    cout<<ans;
}