bzoj1912 [Apio2010]patrol 巡逻（树的直径[变式]）

Description

Input
第一行包含两个整数 n, K(1 ≤ K ≤ 2)。接下来 n – 1行，每行两个整数 a, b， 表示村庄a与b之间有一条道路(1 ≤ a, b ≤ n)。

Output
输出一个整数，表示新建了K 条道路后能达到的最小巡逻距离。

Sample Input
8 1
1 2
3 1
3 4
5 3
7 5
8 5
5 6

Sample Output
11

HINT
10%的数据中，n ≤ 1000, K = 1；
30%的数据中，K = 1；
80%的数据中，每个村庄相邻的村庄数不超过 25；
90%的数据中，每个村庄相邻的村庄数不超过 150；
100%的数据中，3 ≤ n ≤ 100,000, 1 ≤ K ≤ 2。

分析：
先考虑K==1的情况
加了一条边之后，图就变成了一棵环套树
显然我们如果要让巡逻距离尽可能短，那么就要使环上的边尽量多，
（因为不在环上的边都要走两遍）
怎么让环大呢，
就是让添加边变成环之前的那条链尽可能长
这就是树上最长链

树的直径

求法：
两遍dfs，
第一遍dfs找到一个最远点，再从最远点dfs，最后得出的最长dis就是树的直径

那么K==1的时候就是求一个直径dis
ans=2*(n-1-dis)+dis+1

K==2时，就是在K==1求出解得基础上，求一个次长直径

注意

如果我们什么处理都没有，直接求一个次长链（次短路方法），
可能会和最长链重合，那么最长链上的一部分就会走两遍
所以我们在求出最长链之后，把最长链上的边权赋为-1，
这样再跑一个裸的直径就好了
（这样就可以保证可以在新求出的直径中尽量少重合原先的直径）

tip

代码中我用了两种求直径的方法
注意dp返回值是当前点的f值
然而答案要单独统计

这里写代码片
#include<cstdio>
#include<cstring>
#include<iostream>

using namespace std;

const int N=100010;
struct node{
    int x,y,nxt,v;
};
node way[N<<1];
int st[N],tot=-1,n,K;
int len1,len2,pre[N],ansx,ans;
int f[N],g[N];

void add(int u,int w,int z)
{
    tot++;
    way[tot].x=u;way[tot].y=w;way[tot].v=z;way[tot].nxt=st[u];st[u]=tot;
    tot++;
    way[tot].x=w;way[tot].y=u;way[tot].v=z;way[tot].nxt=st[w];st[w]=tot;
}

void dfs(int now,int fa,int dis)
{
    if (dis>ans)
    {
        ans=dis;
        ansx=now;
    }
    for (int i=st[now];i!=-1;i=way[i].nxt)
        if (way[i].y!=fa)
        {
            pre[way[i].y]=i;
            dfs(way[i].y,now,dis+way[i].v);
        }
}

void change(int s,int t)
{
    for (int i=t;i!=s;i=way[pre[i]].x)
    {
        way[pre[i]].v=-1;
        way[pre[i]^1].v=-1;
    }
}

int dp(int now,int fa)
{
    f[now]=0;g[now]=0;
    for (int i=st[now];i!=-1;i=way[i].nxt)
        if (way[i].y!=fa)
        {
            int len=dp(way[i].y,now)+way[i].v;
            if (len>f[now])
            {
                g[now]=f[now];
                f[now]=len;
            }
            else if (len>g[now]) g[now]=len;
        }
    len2=max(len2,f[now]+g[now]);   //统计答案 
    return f[now];  //dp返回的是最长链 
}

int main()
{
    memset(st,-1,sizeof(st));
    scanf("%d%d",&n,&K);
    for (int i=1;i<n;i++)
    {
        int u,w;
        scanf("%d%d",&u,&w);
        add(u,w,1);
    }

    memset(pre,-1,sizeof(pre));
    ans=ansx=0;
    dfs(1,0,0);
    int p=ansx;
    ans=ansx=0;
    memset(pre,-1,sizeof(pre));
    dfs(p,0,0);
    if (K==1)
    {
        printf("%d",2*(n-1-ans)+ans+1);
        return 0;
    }

    len1=ans;
    change(p,ansx);
    dp(1,0);
    printf("%d\n",2*(n-1-len1-len2)+len1+len2+2);
    return 0;
}