hdu 1598 第一道並查集

雖然在學數據結構的時候看過並查集。。。好像是在Kruskal算法中講到了，不過做題中用到這還是頭一次。。。哎，acm有太多的第一次了。。。不過生命不休，學習不止。。。

好吧。。。不多說了。。。

#include<iostream>
#include<algorithm>
const int N=1010;
const int inf=1000000;
using namespace std;

struct Road{
    int st,ed,pd; //起點，終點、速度
};

int cmp(const Road &p,const Road &q){
    return p.pd<q.pd;
}

int parent[N];
int n,m;

//初始化
void UFset(){
    for(int i=1;i<=n;i++){
        parent[i]=-1;
    }
}
//查找並返回x所屬集合的根結點
int Find(int x){
    int s;
    for(s=x;parent[s]>=0;s=parent[s]); //一直查到parent[s]為負數（此時s為根結點)為止
    //路徑壓縮，優化，便於后續的查找操作加速
    while(s!=x){
        int temp=parent[x];
        parent[x]=s;
        x=temp;
    }
    return s;
}
//R1、R2是兩個元素，屬於不同的集合，現在合並這兩個集合
void Union(int R1,int R2){
    int r1=Find(R1); //r1為R1的根結點
    int r2=Find(R2); //r2為R2的根結點
    //路徑為r1->r2,因此根結點r2所在的樹作為r1的子樹
    if(r1!=r2){
        parent[r2]=r1;
    }
}

int main(){
    while(scanf("%d%d",&n,&m)!=EOF){
        Road road[N];
        for(int i=1;i<=m;i++){
            scanf("%d%d%d",&road[i].st,&road[i].ed,&road[i].pd);
        }
    //    sort(road,road+m,cmp);orz//////就是在這兒被坑了好久，我是從1開始的，排序的時候當然得從road[1]開始了。。。。。。教訓那。。。orz
        sort(road+1,road+1+m,cmp);
        int _case;
        scanf("%d",&_case);
        while(_case--){
            int st,ed;
            int min=inf;
            scanf("%d%d",&st,&ed);
            for(int i=1;i<=m;i++){
                UFset();
                for(int j=i;j<=m;j++){
                    Union(road[j].st,road[j].ed); //進行合並
                    //如果在同一集合中、、、
                    if(Find(st)==Find(ed)){
                        int ans=road[j].pd-road[i].pd;  //舒適度吧
                        if(ans<min)
                            min=ans;
                        break;
                    }
                }
            }
            if(min==inf){
                printf("-1\n");
            }else 
                printf("%d\n",min);
        }
    }
    return 0;
}