【bzoj4813】[Cqoi2017]小Q的棋盘 树上dfs+贪心

时间:2023-03-08 22:35:48

题目描述

小Q正在设计一种棋类游戏。在小Q设计的游戏中,棋子可以放在棋盘上的格点中。某些格点之间有连线,棋子只能在有连线的格点之间移动。整个棋盘上共有V个格点,编号为0,1,2…,V-1,它们是连通的,也就是说棋子从任意格点出发,总能到达所有的格点。小Q在设计棋盘时,还保证棋子从一个格点移动到另外任一格点的路径是唯一的。小Q现在想知道,当棋子从格点0出发,移动N步最多能经过多少格点。格点可以重复经过多次,但不重复计数。

输入

第一行包含2个正整数V,N,其中V表示格点总数,N表示移动步数。
接下来V-1行,每行两个数Ai,Bi,表示编号为Ai,Bi的两个格点之间有连线。
V,N≤100, 0 ≤Ai,Bi<V 

输出

输出一行一个整数,表示最多经过的格点数量。

样例输入

5 2
1 0
2 1
3 2
4 3

样例输出

3


题解

树上dfs+贪心

先dfs求出以0为起点的最长一条链,这条链上的点只经过一次,消耗1步;其它的点经过后需要返回这条链上,消耗2步。

然后分类讨论是否能走完链和走完树即可。

#include <cstdio>
#define N 110
int head[N] , to[N << 1] , next[N << 1] , cnt , deep[N];
void add(int x , int y)
{
to[++cnt] = y , next[cnt] = head[x] , head[x] = cnt;
}
void dfs(int x , int fa)
{
int i;
for(i = head[x] ; i ; i = next[i])
if(to[i] != fa)
deep[to[i]] = deep[x] + 1 , dfs(to[i] , x);
}
int main()
{
int n , p , i , x , y , l = 0;
scanf("%d%d" , &n , &p);
for(i = 1 ; i < n ; i ++ )
scanf("%d%d" , &x , &y) , add(x , y) , add(y , x);
dfs(0 , -1);
for(i = 1 ; i < n ; i ++ )
if(l < deep[i])
l = deep[i];
if(p <= l) printf("%d\n" , p + 1);
else if(p >= l + 2 * (n - l - 1)) printf("%d\n" , n);
else printf("%d\n" , l + (p - l) / 2 + 1);
return 0;
}