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;
}