The Flee Plan of Groundhog(DFS)

The Flee Plan of Groundhog(DFS)

思路：因为$n$到每个点所有时间一定，考虑以$n$为根$dfs$，求每个点到$n$的距离$d[i]$

并求出每个结点的$fa[i]$，然后找到$t$秒后$Groundhog$的位置$s$，然后再从$s$开始$dfs$一遍取答案的最大值，再与$d[s]$取较小值。

因为时间最大不能超过$d[s]$，这表示$s$在到达目的地的途中被抓了。

否则取$s$所有能到达的目的地的$\frac{d[i]+1}{2}$即可。

时间复杂度：$O(n)$

#include
using namespace std;
typedef long long ll;
const int N=1e5+5,M=2e4+5,inf=0x3f3f3f3f,mod=1e9+7;
#define mst(a,b) memset(a,b,sizeof a)
#define lx x<<1
#define rx x<<1|1
#define reg register
#define PII pair
#define fi first
#define se second
#define pb push_back
#define il inline
int n,t,s=1,d[N],f[N],ans;
vector<int>e[N];
void dfs(int u,int fa){
	f[u]=fa;
	for(int v:e[u])	if(v!=fa) d[v]=d[u]+1,dfs(v,u);
} 
void dfs1(int u){
	ans=max(ans,(d[u]+1)>>1);
	for(int v:e[u]) if(v!=f[u]) dfs1(v);
}
int main(){
	scanf("%d%d",&n,&t);
	for(int i=1,u,v;i<n;i++) scanf("%d%d",&u,&v),e[u].pb(v),e[v].pb(u);
	dfs(n,0);
	for(int i=1;i<=t;i++) s=f[s];
	dfs1(s);
	printf("%d\n",min(ans,d[s])); 
	return 0;
}