#6WA求助
查看原帖
#6WA求助
767353
Oct0pus1楼主2022/10/7 09:17
#include<cstdio>
#include<vector>
#include<algorithm>
using namespace std;
const int L=1e5+10;
vector<int> e[L];
int n,m,dis[L],f[L],maxdep[L],ans[L];
inline void dfs1(int u,int fa){
	dis[u]=dis[fa]+1;
	f[u]=fa;
	for(int i=0;i<e[u].size();i++){
		int v=e[u][i];
		if(v!=fa)dfs1(v,u);
	}
}
inline void dfs2(int u,int fa){
	dis[u]=dis[fa]+1;
	for(int i=0;i<e[u].size();i++){
		int v=e[u][i];
		if(v!=fa)dfs2(v,u);
	}
	if(e[u].size()==1)maxdep[u]=dis[u];
	maxdep[fa]=max(maxdep[fa],maxdep[u]);
}
int main(){
	scanf("%d%d",&n,&m);
	for(int i=1;i<n;i++){
		int u,v;
		scanf("%d%d",&u,&v);
		e[u].push_back(v);
		e[v].push_back(u);
	}
	dfs1(1,0);
	int maxx=dis[1],p=1,p1=1;
	for(int i=1;i<=n;i++)
		if(dis[i]>maxx){
			maxx=dis[i];
			p=i;
		}
	dfs1(p,0);
	maxx=dis[1];
	for(int i=1;i<=n;i++)
		if(dis[i]>maxx){
			maxx=dis[i];
			p1=i;
		}
	for(int i=1;i<=maxx/2;i++)p1=f[p1];
	dfs2(p1,0);
	for(int i=1;i<=n;i++)ans[i]=maxdep[i]-dis[i];
	sort(ans+1,ans+n+1);
	int pp=n;
	for(int i=1;i<=m;i++)pp--;
	printf("%d",ans[pp-1]+1);
	return 0;
}
2022/10/7 09:17
加载中...