HDU 2874 Connections between cities (LCA)

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=2874

题意是给你n个点，m条边(无向)，q个询问。接下来m行，每行两个点一个边权，而且这个图不能有环路。然后接下来q行，每行给你两个点，问你这两个点的最短距离是多少，要是不相连，则输出一串英文。

首先想到的是用(二分)倍增LCA，但是这题的坑点是两个点可能不在同一个图中，所以我dfs的时候用block[i]标记这个点属于哪一个图中，要是这个点在同一个图中，答案就是cost[u] + cost[v] - 2*cost[lca(u , v)]。

 #include <iostream>
 #include <cstdio>
 #include <cstring>
 #include <vector>
 using namespace std;
 const int MAXN = ;
 struct data {
     int next , to , w;
 }edge[MAXN * ];
 int head[MAXN] , par[MAXN * ][] , dep[MAXN] , cont , cost[MAXN] , block[MAXN];

 void init() {
     cont = ;
     //memset(par , -1 , sizeof(par));
     memset(head , - , sizeof(head));
     memset(block ,  , sizeof(block));
 }

 inline void add(int u , int v , int w) {
     edge[cont].next = head[u];
     edge[cont].to = v;
     edge[cont].w = w;
     head[u] = cont++;
 }

 void dfs(int u , int p , int d , int node , int w) {
     dep[u] = d;
     par[u][] = p;
     block[u] = node;
     cost[u] = w;
     int v;
     for(int i = head[u] ; ~i ; i = edge[i].next) {
         v = edge[i].to;
         if(v == p)
             continue;
         dfs(v , u , d +  , node , w + edge[i].w);
     }
 }

 int lca(int u , int v) {
     if(dep[u] < dep[v])
         swap(u , v);
     for(int k =  ; k <  ; k++) {
         if((dep[u] - dep[v]) >> k & ) {
             u = par[u][k];
         }
     }
     if(u == v)
         return v;
     for(int k =  ; k >=  ; k--) {
         if(par[u][k] != par[v][k]) {
             u = par[u][k];
             v = par[v][k];
         }
     }
     return par[u][];
 }

 int main()
 {
     int n , m , q , u , v , w;
     while(~scanf("%d %d %d" , &n , &m , &q)) {
         init();
         for(int i =  ; i < m ; i++) {
             scanf("%d %d %d" , &u , &v , &w);
             add(u , v , w);
             add(v , u , w);
         }
         int f = ;
         for(int i =  ; i <= n ; i++) {
             if(!block[i]) {
                 dfs(i , - ,  , ++f , );
             }
         }
         for(int k =  ; k <  ; k++) {
             for(int i =  ; i <= n ; i++) {
                 if(par[i][k] <= )
                     par[i][k + ] = par[i][k];
                 else
                     par[i][k + ] = par[par[i][k]][k];
             }
         }
         while(q--) {
             scanf("%d %d" , &u , &v);
             if(block[u] != block[v]) {
                 printf("Not connected\n");
             }
             else {
                 printf("%d\n" , cost[u] + cost[v] -  * cost[lca(u , v)]);
             }
         }
     }
 }