Codefroces Gym 100781A（树上最长路径）

http://codeforces.com/gym/100781/attachments

题意：有N个点，M条边，问对两两之间的树添加一条边之后，让整棵大树最远的点对之间的距离最近，问这个最近距离是多少。

思路：一开始看成只有两个连通块，后来才注意到是多个连通块。DFS搜树上最长路径。答案有三种：第一种是添加了边之后，树的最长路径还是原来子树的路径，第二种是对子树长度进行排序后，两个最长的距离分别除以2向上取整后加上1。第三种比较难注意到，就是第二第三长的分别除以2向上取整后加上2，因为可能隔着一条边之后比第一种情况更长了。

 #include <cstdio>
 #include <algorithm>
 #include <iostream>
 #include <cstring>
 #include <string>
 #include <cmath>
 #include <queue>
 #include <vector>
 using namespace std;
 #define N 100010
 struct node
 {
     int v, nxt;
 }edge[N*];
 int head[N], tot;
 bool vis[N];
 int ans[N];
 int l;

 void add(int u, int v)
 {
     edge[tot].v = v; edge[tot].nxt = head[u]; head[u] = tot++;
 }

 bool cmp(const int &a, const int &b)
 {
     return a > b;
 }

 int dfs(int u)
 {
     vis[u] = ;
     int m1 = , m2 = ;
     for(int i = head[u]; ~i; i = edge[i].nxt) {
         int v = edge[i].v;
         if(vis[v]) continue;
         int tmp = dfs(v) + ;
         if(tmp > m1) {
             m2 = m1, m1 = tmp;
         } else if(tmp > m2) {
             m2 = tmp;
         }
     }
     if((m1 + m2) > l) l = m1 + m2;
     return m1;
 }

 int main()
 {
     int n, m;
     scanf("%d%d", &n, &m);
     memset(vis, , sizeof(vis));
     memset(head, -, sizeof(head));
     memset(ans, , sizeof(ans));
     tot = ;
     for(int i = ; i < m; i++) {
         int u, v;
         scanf("%d%d", &u, &v);
         add(u, v); add(v, u);
     }
     int cnt = , res = ;
     for(int i = ; i < n; i++) {
         if(!vis[i]) {
             l = ;
             dfs(i); //搜树上最长路径
             if(l > res) res = l; //第一种情况
             ans[cnt++] = l;
         }
     }
     sort(ans, ans + cnt, cmp);
     if(cnt > ) res = max(res, (ans[] + ) /  + (ans[] + ) /  + ); //第二种情况
     if(cnt > ) res = max(res, (ans[] + ) /  + (ans[] + ) /  + ); //第三种情况
     printf("%d\n", res);
     return ;
 }