随笔 - 531 文章 - 0 评论 - 3 阅读 - 10215

P3629 巡逻

在边权都为1的树上加1条或者2条边（边权为1) ，从1出发遍历所有点，使得总路程最短

#include <bits/stdc++.h>
using namespace std ;
 const int N=1e5+2,M=2*N;
 int nxt[M],w[M],go[M],hd[N],all=1;
 int D,d1[N],d2[N],b[N],m,n,id,d[N],ans,pre[N];
  
 void add(int x,int y,int z){
    go[++all]=y,w[all]=z,nxt[all]=hd[x],hd[x]=all;
 }
 void dfs1(int x,int last){
    int i,y,z;
    if(d[x]>ans){
        ans=d[x],id=x;
    }
    for(i=hd[x];i;i=nxt[i]){
        y=go[i],z=w[i]; 
         if(y!=last){
            d[y]=d[x]+z; pre[y]=i; dfs1(y,x);   
         }
    }
 }
 void dfs(int x,int last){
    int i,y,z;
    for(i=hd[x];i;i=nxt[i]){
        y=go[i],z=w[i]; 
        if(y!=last){
            dfs(y,x);
            if(d1[x]<d1[y]+z) d2[x]=d1[x],d1[x]=d1[y]+z;
            else if(d2[x]<d1[y]+z) d2[x]=d1[y]+z;
        }
    }
    D=max(D,d1[x]+d2[x]);
 }
 signed main(){
    int i,t,x,y,z;
    cin>>n>>m;
    for(i=1;i<n;i++) cin>>x>>y,add(x,y,1),add(y,x,1);
    dfs1(1,0); t=id;
    memset(d,0,sizeof(d)),ans=0; dfs1(id,0);
     
    if(m==1){
        cout<<2*(n-1)-ans+1;
        return 0;
    }
    x=id;
    for(;x!=t;x=go[i^1]){
        i=pre[x]; w[i]=w[i^1]=-1; 
    }
    D=0;
    dfs(1,0);
    cout<<n*2-ans-D;
    return 0;
 }