易错点：

• l2必须要设置全局变量而不能设置局部变量，这是由于设置局部变量无法兼顾所有情况造成的.
• bfs并设置直径上边权为-1后，如果k=2则不能继续直接使用bfs获得答案，这是bfs的拓展加点性造成的（类比dijkstra）.
  如果两个变量可以用一个变量导出就用一个变量.

BFS方法正确性的证明：

• 如果源点在直径上：显然正确.
• 如果源点不在直径上：既然源点不在直径上那么直径一定在源点的某棵子树内。由树和直径的性质可知此时直接取最远点然后再进行一次bfs即可求得直径.

#include<cstdio>
#include<iostream>
#include<queue>
#include<cstring>
using namespace std;
const int MAXN=2e5,INF=0x3f3f3f3f;
struct Edge{
    int from,to,w,nxt;
}e[MAXN*2];
int head[MAXN],edgeCnt=1;
void addEdge(int u,int v,int w){
    e[++edgeCnt].from=u;
    e[edgeCnt].to=v;
    e[edgeCnt].w=w;
    e[edgeCnt].nxt=head[u];
    head[u]=edgeCnt;
}
int n;
int d[MAXN],pre[MAXN];
int bfs(int p){
    memset(d,0x3f,sizeof(d));
    queue<int> q;
    q.push(p);
    d[p]=pre[p]=0;
    while(!q.empty()){
        int nowNode=q.front();q.pop();
        for(int i=head[nowNode];i;i=e[i].nxt){
            int nowV=e[i].to;
            if(d[nowV]==INF){
                d[nowV]=d[nowNode]+e[i].w;
                pre[nowV]=i;
                q.push(nowV);
            }
        }
    }
    int ans=1;
    for(int i=1;i<=n;i++){
        if(d[i]>d[ans])ans=i;//如果两个变量可以用一个变量导出就用一个变量. 
    }
    return ans;
}
int p;
int get(){
    p=bfs(1);
    p=bfs(p);
    return d[p];
}
void change(){
    for(;pre[p];p=e[pre[p]].from){
        e[pre[p]].w=e[pre[p]^1].w=-1;
    }
}
bool vis[MAXN];
int f[MAXN],l2;
void dfs(int p){
    vis[p]=1;
    for(int i=head[p];i;i=e[i].nxt){
        int nowV=e[i].to;
        if(!vis[nowV]){
            dfs(nowV);
            l2=max(l2,f[p]+f[nowV]+e[i].w);
            f[p]=max(f[p],f[nowV]+e[i].w);
        }
    }
}
int main(){
    int k;
    scanf("%d%d",&n,&k);
    for(int i=1;i<=n-1;i++){
        int a,b;
        scanf("%d%d",&a,&b);
        addEdge(a,b,1);
        addEdge(b,a,1);
    }
    int l1=get();
    if(k==1){
        printf("%d\n",2*(n-1)-l1+1);
        return 0;
    }else{
        change();
        dfs(1);
        printf("%d\n",2*n-l1-l2);
        return 0;
    }
}