APIO2010 巡逻 树的直径

题目大意：给定一棵树，加上1或者2条边，求在加边后，从 1 号节点出发遍历所有边再回到 1 号点的最小距离是多少。加上的边必须恰好经过一次。
题解：

• 不建立新的道路时，根据DFS遍历思想可以证明每条边都要走2次，总巡逻长度为2*(n - 1)。
• 建立一条新道路时，被这条道路所覆盖的边就只需要经过一次了（因为形成了环）。所以让这条边覆盖树的直径，答案为2(n - 1) - d1 +１。
• 建立两条新道路时，又会形成一个环。当两个环重叠的时候，重叠部分又要经过2次了（可以自己模拟一下）。所以我们把直径的边权设为-1，再求一次直径。此时答案为2*(n - 1) - d1 + 1 - d2 + 1 = 2*n - d1 - d2。
• 第二次求直径时，由于边权不全为1了，所以只能用DP求出直径长度。

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int MAXN = 100001;
const int INF = 0x7fffffff;

int fir[MAXN], nxt[MAXN << 1], to[MAXN << 1], len[MAXN << 1], cnt;
int d1, d2;
int dis[MAXN], Begin, End, last[MAXN][2];

inline int read(){
    int k = 0, f = 1; char ch = getchar();
    while(ch < '0' || ch > '9'){if(ch == '-') f = -1; ch = getchar();}
    while(ch >= '0' && ch <= '9'){k = k*10 + ch - '0'; ch = getchar();}
    return k * f;
}

inline void add_edge(int a, int b, int l){
    len[cnt] = 1;
    to[cnt] = b;
    nxt[cnt] = fir[a];
    fir[a] = cnt++;
}

void dfs1(int u, int fa){
    if(dis[u] > dis[Begin])
        Begin = u;
    for(int i = fir[u]; i != -1; i = nxt[i]){
        int v = to[i];
        if(v == fa) continue;
        dis[v] = dis[u] + len[i];
        dfs1(v, u);
    }
}

void dfs2(int u, int fa){
    if(dis[u] > dis[End] && u != Begin)
        End = u;
    for(int i = fir[u]; i != -1; i = nxt[i]){
        int v = to[i];
        if(v == fa) continue;
        dis[v] = dis[u] + len[i];
        last[v][0] = u, last[v][1] = i;
        dfs2(v, u);
    }
}

void dfs3(int u, int fa){
    for(int i = fir[u]; i != -1; i = nxt[i]){
        int v = to[i];
        if(v == fa) continue;
        dfs3(v, u);
        d2 = max(d2, dis[u] + dis[v] + len[i]);
        dis[u] = max(dis[u], dis[v] + len[i]);
    }
}

int Solve_Diameter(){ //返回树的直径的长度 
    Begin = End = 0;
    memset(dis, 0, sizeof(dis)); dis[Begin] = -INF;
    dfs1(1, 0);
    memset(dis, 0, sizeof(dis)); dis[End] = -INF;
    dfs2(Begin, 0);
    return dis[End];
}

int main(){
    freopen("in.txt", "r", stdin);
    memset(fir, -1, sizeof(fir));

    int n = read(), k = read();
    for(int i = 1; i < n; i++){
        int a = read(), b = read();
        add_edge(a, b, 1), add_edge(b, a, 1);
    }

    d1 = Solve_Diameter();
//  printf("d1 = %d, Begin = %d, End = %d\n", d1, Begin, End);
    if(k == 1){
        printf("%d", 2 * n - d1 - 1);
        return 0;//2 * (n - 1) - d1 + 1
    }
    for(int i = End; i != Begin; i = last[i][0]){
        len[last[i][1]] = len[last[i][1] ^ 1] = -1;
    }

    memset(dis, 0, sizeof(dis));
    dfs3(1, 0);
//  printf("d2 = %d\n", d2);
    printf("%d", 2 * n - d1 - d2);
    return 0; 
}