题目传送门
在最初的树上,求出树的直径L1 然后将这条路径上的所有边权统统取反.
1变成−1
我们再求一次树的直径L2L2.
答案就是
2×(n−1)−(L1−1)−(L2−1)=2×(n−1)−L1+1−L2+1
假如说L2L2和L1L1有重叠部分.
那么当我们
−L1+1
的时候,我们就会发现,重叠的部分变成了只需要经过一次.
然后
−L2+1
−L2+1
相当于把重叠部分相加回来了.
此时变成了经过了两次.
假如说没有重叠部分,那么1,-1都不会有影响,反正我们不会经过这些边.
#include<cstdio>
#include<iostream>
#include<cstring>
#include<queue>
using namespace std;
const int MAXX=100005;
queue<int>q;
int h[MAXX],e[MAXX*2],w[MAXX*2],nxt[MAXX*2];
int d[MAXX],pre[MAXX],f[MAXX];
bool v[MAXX];
int n,k,p,tot=1,ans;
void add(int x,int y,int z){
e[++tot]=y;
nxt[tot]=h[x];
h[x]=tot;
w[tot]=z;
}
int bfs(int s){
memset(d,0x3f,sizeof(d));
d[s]=0;
pre[s]=0;
q.push(s);
while(!q.empty()){
int x=q.front();
q.pop();
for(int i=h[x];i!=-1;i=nxt[i]){
int y=e[i];
if(d[y]==0x3f3f3f3f){
pre[y]=i;
d[y]=d[x]+w[i];
q.push(y);
}
}
}
int x,y;
for( x=y=1;x<=n;x++){
if(d[x]>d[y])y=x;
}//找到端点
return y;
}
void dp(int x){
v[x]=1;
for(int i=h[x];i!=-1;i=nxt[i]){
int y=e[i];
if(v[y])continue;
dp(y);
ans=max(ans,f[x]+f[y]+w[i]);
f[x]=max(f[x],f[y]+w[i]);
}
}
void change(){
for(;pre[p];p=e[pre[p]^1])w[pre[p]]=w[pre[p]^1]=-1;
//取反操作,ver[pre[p]^1],实际上是反边的终点,也就是p的前继
//因为tot=1;++tot,边从2开始记录,2^1=3,即2,3互为反边
}
int main(){
memset(h,-1,sizeof h);
cin>>n>>k;
for(int i=1;i<=n-1;i++){
int x,y;
cin>>x>>y;
add(x,y,1);
add(y,x,1);
}
p=bfs(1);
p=bfs(p);
int l1=d[p];
if(k==1){
cout<<2*(n-1)-(l1-1)<<endl;
}else {
change();
dp(1);
cout<<2*(n-1)-(l1-1)-(ans-1)<<endl;
}
return 0;
}