【bzoj 1912】patrol 巡邏（樹DP）

傳送門biu~
在加新路前每條邊需要走兩次，加一條新路(u,v)可以使總距離減小dis(u,v)。所以在k=1時只需求樹的直徑；在k=2時選兩條路的路徑長度和最大，將選擇的兩條路(u1,v1)、(u2,v2)在樹上相交的部分抵消掉，可以在求完第一條最長路以後將這條路上的權值都改成-1，這樣在跑第二條路的時候就可以看作相互抵消了。

#include<bits/stdc++.h>
#define N 100050
using namespace std;
int n,k,Ans,ans,Root,Son,P;
int head[N],nex[N<<1],to[N<<1],w[N<<1],tp;
int f[N][2],g[N],p[N];
inline void add(int x,int y){
    nex[++tp]=head[x];
    head[x]=tp;
    to[tp]=y;
    w[tp]=1;
}
void dfs(int x,int father){
    int now,son=0,pp;
    for(int i=head[x];i;i=nex[i]){
        if(to[i]==father)   continue;
        dfs(to[i],x);   now=f[to[i]][0]+w[i];
        if(now>f[x][0]){
            son=g[x];
            pp=p[x];
            f[x][1]=f[x][0];
            f[x][0]=now;
            g[x]=to[i];
            p[x]=i;
        }
        else if(now>f[x][1]){
            f[x][1]=now;
            son=to[i];
            pp=i;
        }
    }
    if(f[x][0]+f[x][1]>ans){
        ans=f[x][0]+f[x][1];
        Root=x; Son=son; P=pp;
    }
}
int main(){
    scanf("%d%d",&n,&k);
    for(int i=1;i<n;++i){
        int x,y;
        scanf("%d%d",&x,&y);
        add(x,y);add(y,x);
    }
    dfs(1,0);
    Ans=2*(n-1)-ans+1;
    if(k==1){
        printf("%d\n",Ans);
        return 0;
    }
    if(f[Root][1]){
        int x=Son;
        w[P]=-1;
        while(g[x]) w[p[x]]=-1,x=g[x];
    }
    int x=Root; while(g[x]) w[p[x]]=-1,x=g[x];
    ans=0;  memset(f,0,sizeof f);   dfs(1,0);
    Ans-=ans-1; printf("%d\n",Ans);
    return 0;
}