hdu 1598 第一道并查集

虽然在学数据结构的时候看过并查集。。。好像是在Kruskal算法中讲到了，不过做题中用到这还是头一次。。。哎，acm有太多的第一次了。。。不过生命不休，学习不止。。。

好吧。。。不多说了。。。

View Code

 1 #include<iostream>
 2 #include<algorithm>
 3 const int N=1010;
 4 const int inf=1000000;
 5 using namespace std;
 6 
 7 struct Road{
 8     int st,ed,pd; //起点，终点、速度
 9 };
10 
11 int cmp(const Road &p,const Road &q){
12     return p.pd<q.pd;
13 }
14 
15 int parent[N];
16 int n,m;
17 
18 //初始化
19 void UFset(){
20     for(int i=1;i<=n;i++){
21         parent[i]=-1;
22     }
23 }
24 //查找并返回x所属集合的根结点
25 int Find(int x){
26     int s;
27     for(s=x;parent[s]>=0;s=parent[s]); //一直查到parent[s]为负数（此时s为根结点)为止
28     //路径压缩，优化，便于后续的查找操作加速
29     while(s!=x){
30         int temp=parent[x];
31         parent[x]=s;
32         x=temp;
33     }
34     return s;
35 }
36 //R1、R2是两个元素，属于不同的集合，现在合并这两个集合
37 void Union(int R1,int R2){
38     int r1=Find(R1); //r1为R1的根结点
39     int r2=Find(R2); //r2为R2的根结点
40     //路径为r1->r2,因此根结点r2所在的树作为r1的子树
41     if(r1!=r2){
42         parent[r2]=r1;
43     }
44 }
45 
46 int main(){
47     while(scanf("%d%d",&n,&m)!=EOF){
48         Road road[N];
49         for(int i=1;i<=m;i++){
50             scanf("%d%d%d",&road[i].st,&road[i].ed,&road[i].pd);
51         }
52     //    sort(road,road+m,cmp);orz//就是在这儿被坑了好久，我是从1开始的，排序的时候当然得从road[1]开始了。。。。。。教训那。。。orz
53         sort(road+1,road+1+m,cmp);
54         int _case;
55         scanf("%d",&_case);
56         while(_case--){
57             int st,ed;
58             int min=inf;
59             scanf("%d%d",&st,&ed);
60             for(int i=1;i<=m;i++){
61                 UFset();
62                 for(int j=i;j<=m;j++){
63                     Union(road[j].st,road[j].ed); //进行合并
64                     //如果在同一集合中、、、
65                     if(Find(st)==Find(ed)){
66                         int ans=road[j].pd-road[i].pd;  //舒适度吧
67                         if(ans<min)
68                             min=ans;
69                         break;
70                     }
71                 }
72             }
73             if(min==inf){
74                 printf("-1\n");
75             }else 
76                 printf("%d\n",min);
77         }
78     }
79     return 0;
80 }