省选专练[USACO18JAN]Cow at Large Gold

最新推荐文章于 2023-05-30 21:58:44 发布

原创最新推荐文章于 2023-05-30 21:58:44 发布 · 278 阅读

1 ·

CC 4.0 BY-SA版权

动态规划同时被 3 个专栏收录

39 篇文章

订阅专栏

USACO

14 篇文章

订阅专栏

2018

12 篇文章

订阅专栏

本文介绍了一种使用两次深度优先搜索(DFS)实现的简单树形动态规划(DP)算法，该算法首先计算节点的深度及到最近叶子节点的距离，然后找出所有深度小于到最近叶子节点距离的节点数量。

简单的树形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;
}