[luogu]P3629 [APIO2010]巡逻

#include using namespace std;
const int N=1e5+1000;
int read(){
    char c;int num,f=1;
    while(c=getchar(),!isdigit(c))if(c==‘-‘)f=-1;num=c-‘0‘;
    while(c=getchar(), isdigit(c))num=num*10+c-‘0‘;
    return f*num;
}
int n,k,ans,pre[N*2],maxn,id,dis[N*2];
int head[N],nxt[N*2],ver[N*2],edge[N*2],tot=1;
void add_edge(int u,int v,int w){
    ver[++tot]=v;edge[tot]=w;nxt[tot]=head[u];head[u]=tot;
    ver[++tot]=u;edge[tot]=w;nxt[tot]=head[v];head[v]=tot;
}
void dfs(int x,int val,int pr){
    if(val>maxn){
        maxn=val;
        id=x;
    }
    for(int i=head[x];i;i=nxt[i]){
        if(ver[i]==pr)continue;
        pre[ver[i]]=i;
        dfs(ver[i],val+edge[i],x);
    }
}
int dfs_get_diam(){
    dfs(1,0,1);
    memset(pre,0,sizeof(pre));
    maxn=0;
    dfs(id,0,id);
    return maxn;
}
void dp_get_diam(int x,int fa){
    for(int i=head[x];i;i=nxt[i]){
        if(ver[i]==fa)continue;
        dp_get_diam(ver[i],x);
        maxn=max(maxn,dis[x]+dis[ver[i]]+edge[i]);
        dis[x]=max(dis[x],dis[ver[i]]+edge[i]);
    }
}
int main()
{
    n=read();k=read();
    for(int i=1;i){
        int u,v;
        u=read();v=read();
        add_edge(u,v,1);
        ans+=2;
    }
    ans-=dfs_get_diam()-1;
    if(k==1){
        printf("%d\n",ans);
        return 0;
    }
    for(int i=pre[id];i;i=pre[ver[i^1]]){
        edge[i]=-1;
        edge[i^1]=-1;
    }
    maxn=0;
    dp_get_diam(1,1);
    ans-=maxn-1;
    printf("%d\n",ans);
    return 0;
}