HDU-1598-find the most comfortable road(暴力枚举+Kruskal最小生成树)

题意：

给出一个图，然后Q个询问，每次询问从一个节点到另一个节点，联通图中的“最大边和最小边之差”的最小值，但如果节点之间不连通，则输出-1.

思路：由于询问Q < 11，m < 1000，所以O（Q*n^2），Q*n^2 < 10^8，用最小生成树的思路，在给图上的边排好序的基础上，每次枚举最小边，然后做并查集枚举最大边，当查询的路径相通的时候，如果差值比上一次枚举最小边的差值小的时候，更新。

AC代码：

//AC
#include
#include
#include
#include
using namespace std;
#define maxn 20500
#define INF 9000000
int par[maxn], h[maxn], n, m, q;
struct node{
   int s, t, w;
};
bool cmp(node a, node b){
    return a.w < b.w;
}
vector ve;
void init(){
    for(int i = 0; i <= n; i++){
       par[i] = i;
       h[i] = 0;
    }
}

int Find(int x){
   if(par[x] == x) return x;
   return par[x] = Find(par[x]);
}

void Union(int a, int b){
   a = Find(a);
   b = Find(b);
   if(a != b){
     if(h[a] > h[b]) par[b] = par[a];
     else{
         if(h[a] == h[b]) h[b]++;
         par[a] = par[b];
     }
   }
}
int query(int a, int b){
    int ans = INF, j;
    for(int i = 0; i < ve.size(); i++){//枚举最小边
         init();
         for(j = i ; j < ve.size(); j++){
             int x = Find(ve[j].s), y = Find(ve[j].t);
             if(x != y){
                Union(x, y);
             }
             if(Find(a) == Find(b)) {  //找到最大边
                 if(ans > ve[j].w - ve[i].w)
                    ans = ve[j].w - ve[i].w;
                 break;
             }
         }
         if(j == m) break;  //如果j == m则不必继续往下枚举了
    }
    return ans;
}

void work(){
    ve.clear();
    init();
    for(int i = 0; i < m; i++){
        node n;
        scanf("%d%d%d", &n.s, &n.t, &n.w);
        ve.push_back(n);
    }
    sort(ve.begin(), ve.end(), cmp);
    scanf("%d", &q);
    for(int i = 0; i < q; i++){
       int a, b, c;
       scanf("%d%d", &a, &b);
       c = query(a, b);
       printf("%d\n", (c == INF) ? -1 : c);
    }
}
int main()
{
   while(scanf("%d%d", &n, &m) != EOF){
      init();
      work();
   }
   return 0;
}