这道题是一个树的直径的一道变异板子题。
首先去分析题目,每一个点到核心城市的距离是它到其中某一个的最小值,二我们要去找的最小值,就是在k个城市不断变化的过程中的所有距离的最大值的最小值
其实题目会让你想到erf,但是没有办法去做,可以想到二分的复杂度严重超标,所以我们去看一看题目的性质,所有点距离的最大值,嗯。。。是不是不难想到这个最大值一定是在直径上,因为假如有一个点到某一个点的距离大于直径上的点,那么这个点一定是在直径上。
所以我们就可以想到去在直径上做文章,这k个核心城市的核心是不是一定在这条直径的中点上,假如不是的话(就是不包含直径的中点)那么这个解的值一定会比答案要大的,这个也可以感性的理解一下,最后我们就可以从这个中点dfs,找到每一个点最远可以到达多深,最后用一个前缀和的思想maxdep[x]-dep[x]+1 显然取走最大的k个,最后的k+1个就是答案
代码如下
#include<bits/stdc++.h>
using namespace std;
void read(int &x){
int r=0;
char s;
s=getchar();
while(s<'0'||s>'9') s=getchar();
while(s>='0'&&s<='9'){
r=r*10+s-'0';
s=getchar();
}
x=r;
}
const int maxn=100010;
int n,k;
int head[2*maxn],to[2*maxn],nex[2*maxn];
int num=0;
void add(int x,int y){
to[++num]=y;
nex[num]=head[x];
head[x]=num;
}
int whi;
int dis[maxn];
int maxd=-1;
void dfs1(int x,int f){
if(dis[x]>maxd){
maxd=dis[x];
whi=x;
}
for(int i=head[x];i;i=nex[i]){
if(to[i]==f) continue;
dis[to[i]]=dis[x]+1;
dfs1(to[i],x);
}
}
int fa[maxn];
void dfs2(int x,int f){
fa[x]=f;
if(dis[x]>maxd){
maxd=dis[x];
whi=x;
//cout<<whi<<endl;
}
for(int i=head[x];i;i=nex[i]){
if(to[i]==f) continue;
dis[to[i]]=dis[x]+1;
dfs2(to[i],x);
}
}
int low[maxn];
void dfs3(int x,int f){
low[x]=dis[x];
for(int i=head[x];i;i=nex[i]){
if(to[i]==f) continue;
dis[to[i]]=dis[x]+1;
dfs3(to[i],x);
low[x]=max(low[x],low[to[i]]);//最远可以到达哪块
}
}
bool cmp(int x,int y){
return x>y;
}
int main(){
scanf("%d%d",&n,&k);
for(int i=1;i<n;i++){
int a,b;
scanf("%d%d",&a,&b);
add(a,b);
add(b,a);
}
dfs1(1,0);
memset(dis,0,sizeof(dis));
maxd=-1;
dfs2(whi,0);
int kkk=whi;//注意之后的whi会改变,所以要记录
for(int i=1;i<=(dis[whi]+1)/2;i++) kkk=fa[kkk];//找直径中点
memset(dis,0,sizeof(dis));
dfs3(kkk,0);
int dd[maxn];
for(int i=1;i<=n;i++) dd[i]=low[i]-dis[i];//找到这个点最远的距离
sort(dd+1,dd+1+n,cmp);
int ans=0;
for(int i=k+1;i<=n;i++) ans=max(ans,dd[i]+1);
printf("%d",ans);
return 0;
}
//1.树的直径
//2。在循环中加入要改变某一个变量的值,那么他是否可以做这个循环的条件要去考虑 ,,就像这里面去找直径中点的那个操作