比赛 2025.12.13 评测结果 AAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
题目名称 巡逻 最终得分 100
用户昵称 梦那边的美好TE 运行时间 0.663 s
代码语言 C++ 内存使用 6.23 MiB
提交时间 2025-12-13 08:46:12
显示代码纯文本
#include <bits/stdc++.h>
#define mp make_pair
#define pii pair<int,int>
using namespace std;
const int N=2e6+10;
int ver[N],to[N<<1],nxt[N<<1],val[N<<1],idx=1;
int n,k,s,res,d[N],pat[N],ans,rt;
map<pii,bool>H;
inline void read(int &x){
	x=0;int f=1;char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
	while(ch>='0'&&ch<='9')x=(x<<3)+(x<<1)+ch-'0',ch=getchar();
	x*=f;return;
} 
void add(int x,int y){
	to[++idx]=y,nxt[idx]=ver[x],ver[x]=idx,val[idx]=1;
}
void dfs1(int x,int fa){
	if(d[x]>d[s])s=x;
	pat[x]=fa;
	for(int i=ver[x];i;i=nxt[i]){
		int y=to[i];
		if(y==fa)continue;
		d[y]=d[x]+1;
		dfs1(y,x);
	}
	return;
}
void dfs2(int x,int fa){
	d[x]=0;
	for(int i=ver[x];i;i=nxt[i]){
		int y=to[i];
		if(y==fa)continue;
		dfs2(y,x);
		res=max(res,d[x]+d[y]+1-H[mp(x,y)]*2);
		d[x]=max(d[x],d[y]+1-H[mp(x,y)]*2);
	}
	return;
}
int main(){
	freopen("xunluo.in","r",stdin);
	freopen("xunluo.out","w",stdout);
	read(n),read(k);
	ans=2*n-2;
	for(int i=1,u,v;i<n;i++){
		read(u),read(v);
		add(u,v),add(v,u);
	}
	dfs1(1,0);
	d[rt=s]=0;
	dfs1(rt,0);
	ans=ans-d[s]+1;
	if(k==1){
		printf("%d\n",ans);
		return 0;
	}
	while(s!=rt){
		H[mp(s,pat[s])]=1;
		H[mp(pat[s],s)]=1;
		s=pat[s]; 
	}
	dfs2(rt,0);
	ans=ans-res+1;
	printf("%d\n",ans);
	return 0;
	
}