Description
小Q正在设计一种棋类游戏。在小Q设计的游戏中,棋子可以放在棋盘上的格点中。某些格点之间有连线,棋子只能
在有连线的格点之间移动。整个棋盘上共有V个格点,编号为0,1,2…,V-1,它们是连通的,也就是说棋子从任意格
点出发,总能到达所有的格点。小Q在设计棋盘时,还保证棋子从一个格点移动到另外任一格点的路径是唯一的。
小Q现在想知道,当棋子从格点0出发,移动N步最多能经过多少格点。格点可以重复经过多次,但不重复计数。
Input
第一行包含2个正整数V,N,其中V表示格点总数,N表示移动步数。
接下来V-1行,每行两个数Ai,Bi,表示编号为Ai,Bi的两个格点之间有连线。
V,N≤ 100, 0 ≤Ai,Bi<V
Output
输出一行一个整数,表示最多经过的格点数量。
枚举最后停在某个点x,贪心让x到0的路径上的边只经过一次,其余边则经过两次或零次,把剩余步数尽可能用完即可,经过的点数是边数+1
时间复杂度O(n)
#include<cstdio> const int N=107; int n,m,ans=0; int es[N*2],enx[N*2],e0[N],ep=2,dep[N]; void maxs(int&a,int b){if(a<b)a=b;} void f1(int w,int pa){ if(dep[w]>m)return; maxs(ans,dep[w]+m+2>>1); for(int i=e0[w];i;i=enx[i]){ int u=es[i]; if(u!=pa){ dep[u]=dep[w]+1; f1(u,w); } } } int main(){ scanf("%d%d",&n,&m); for(int i=1,a,b;i<n;++i){ scanf("%d%d",&a,&b); es[ep]=b;enx[ep]=e0[a];e0[a]=ep++; es[ep]=a;enx[ep]=e0[b];e0[b]=ep++; } f1(0,-1); if(ans>n)ans=n; printf("%d ",ans); return 0; }