luogu P3698 [CQOI2017]小Q的棋盘

题面传送门
貌似弱省省选不咋地。
考虑贪心，折返走要两倍的步数，而最后一次仅需一倍的步数。
首先\(dfs\)求出最长链，然后将步数尽量给最长链，剩下的步数给往返走。
注意不能超过\(n\)。
代码实现:

#include<cstdio>
#include<cstring>
#define max(a,b) ((a)>(b)?(a):(b))
#define min(a,b) ((a)<(b)?(a):(b))
using namespace std;
int n,m,k,x,y,head,h[100039],ans,tot,pus;
struct yyy{
	int to,z;
}f[100039],tmp;
inline void add(int x,int y){
	f[++head]=(yyy){y,h[x]};
	h[x]=head;
}
inline void dfs(int x,int y,int z){
	ans=max(z,ans);
	int cur=h[x];
	yyy tmp;
	while(cur!=-1){
		tmp=f[cur];
		if(tmp.to!=y)dfs(tmp.to,x,z+1);
		cur=tmp.z;
	}
}
int main(){
	register int i;
	memset(h,-1,sizeof(h));
	scanf("%d%d",&n,&m);
	for(i=1;i<n;i++) scanf("%d%d",&x,&y),add(x,y),add(y,x);
	dfs(0,0,0);
	printf("%d\n",min(min(ans,m)+max(m-ans,0)/2+1,n));
}