牛客多校第九场K-The Flee Plan of Groundhog(树形DP)

Description

牛客多校第九场K-The Flee Plan of Groundhog(树形DP)_第1张图片

Solution

  • t t t t t t秒后Groundhog追到的地方
  • ∴ d i s ( x , t ) = d i s ( x , n ) 2 \therefore dis(x,t)=\cfrac{dis(x,n)}{2} dis(x,t)=2dis(x,n)时被追到
  • m d i md_i mdi i i i周围离 n n n最远的点
  • ∴ a n s = min ⁡ ( d e p x , m d x + 1 2 ) \therefore ans=\min(dep_x,\cfrac{md_x+1}{2}) ans=min(depx,2mdx+1)
#include 
const int N=1e5+10;
int n,t,tot,h[N],d[N],f[N],md[N];
struct Edge{int v,nxt;}e[N<<1];
void add(int u,int v){e[++tot].v=v,e[tot].nxt=h[u],h[u]=tot;}
void dfs1(int x,int fx,int dep){
	f[x]=fx,d[x]=dep,md[x]=dep;
	for(int i=h[x],v;~i;i=e[i].nxt)if((v=e[i].v)^fx)dfs1(v,x,dep+1),md[x]=std::max(md[x],md[v]);
}int main(){
	memset(h,-1,sizeof(h));
	scanf("%d%d",&n,&t);
	for(int i=1,a,b;i<n;++i) scanf("%d%d",&a,&b),add(a,b),add(b,a);
	dfs1(n,0,0);
	int x=1;while(t--){x=f[x];if(x==n)break;}
	printf("%d\n",std::min(d[x],(md[x]+1)>>1));
}

你可能感兴趣的:(2020牛客暑期多校训练营)