2020牛客多校第九场 K-The Flee Plan of Groundhog

链接

题意

给定一颗n个节点的树，开始时Groundhog在1号节点，Orange在n号节点，在前t秒，Groundhog会走向Orange，t秒后，Groundhog会尽量远离Orange，而此时Orange也会尽力追捕Groundhog。Groundhog的速度是1秒走1格，Orange的速度是1秒走2格。

思路

先搜一遍Groundhog到Orange的路径，算出Groundhog在t秒的位置，再根据一遍对Orange的路dfs（bfs一样的）一遍，算出其到任意一点的最快时间，再对Groundhog走的路进行bfs，时间最长的条件下二者时间相同的点即为答案。
（本题数据不强，直接t秒后往远离n的方向走都可以过）

代码

#include<bits/stdc++.h>
#define N 100005
#define re register
#define ll long long
using namespace std;
int n,m,K;
inline void Rd(int &res){
	re char c;res=0;
	while(c=getchar(),c<48);
	do res=(res<<3)+(res<<1)+(c^48);
	while(c=getchar(),c>47);
}
int head[N],to[N<<1],nxt[N<<1],cnt1;
inline void add(int x,int y){to[++cnt1]=y;nxt[cnt1]=head[x];head[x]=cnt1;}
int T,fa[N],dep[N],V[N];
int Q[N],V1[N];
bool mk[N];
void dfs(int x,int f){
	fa[x]=f;dep[x]=dep[f]+1;
	V[x]=dep[x]>>1;
	for(re int i=head[x];i;i=nxt[i]){
		int y=to[i];
		if(y==f)continue;
		dfs(y,x);
	}
}
int main(){
//	freopen("escape.in","r",stdin);
	Rd(n);Rd(T);
	for(re int i=1;i<n;i++){
		int x,y;
		Rd(x),Rd(y);
		add(x,y);
		add(y,x);
	}
	dfs(n,0);
	int res=1;
	while(T--)res=fa[res];
	int l=0,r=1,ans=0;
	Q[1]=res;
	while(l<r){
		int x=Q[++l];
		if(mk[x])continue;
		mk[x]=1;
		if(V1[x]<=V[x])ans=max(ans,V[x]);
		if(V1[x]>=V[x])continue;
		for(re int i=head[x];i;i=nxt[i]){
			int y=to[i];
			if(mk[y])continue;
			V1[y]=V1[x]+1;
			Q[++r]=y;
		}
	}
	printf("%d\n",ans);
	return 0;
}