HDU-1875 畅通工程再续（最小生成树+判断是否存在）

http://acm.hdu.edu.cn/showproblem.php?pid=1875

Problem Description

相信大家都听说一个“百岛湖”的地方吧，百岛湖的居民生活在不同的小岛中，当他们想去其他的小岛时都要通过划小船来实现。现在政府决定大力发展百岛湖，发展首先要解决的问题当然是交通问题，政府决定实现百岛湖的全畅通！经过考察小组RPRush对百岛湖的情况充分了解后，决定在符合条件的小岛间建上桥，所谓符合条件，就是2个小岛之间的距离不能小于10米，也不能大于1000米。当然，为了节省资金，只要求实现任意2个小岛之间有路通即可。其中桥的价格为 100元/米。

Input

输入包括多组数据。输入首先包括一个整数T(T <= 200)，代表有T组数据。
每组数据首先是一个整数C(C <= 100),代表小岛的个数，接下来是C组坐标，代表每个小岛的坐标，这些坐标都是 0 <= x, y <= 1000的整数。

Output

每组输入数据输出一行，代表建桥的最小花费，结果保留一位小数。如果无法实现工程以达到全部畅通，输出”oh!”.

Sample Input

Sample Output

1414.2
oh!

 #include <stdio.h>
 #include <string.h>
 #include <iostream>
 #include <string>
 #include <math.h>
 #include <algorithm>
 #include <vector>
 #include <stack>
 #include <queue>
 #include <set>
 #include <map>
 #include <sstream>
 const int INF=0x3f3f3f3f;
 typedef long long LL;
 const int mod=1e9+;
 //const double PI=acos(-1);
 #define Bug cout<<"---------------------"<<endl
 const int maxn=1e5+;
 using namespace std;

 struct edge_node
 {
     int to;
     double val;
     int next;
 }Edge[maxn];
 int Head[];
 int tot;
 double MST;//存放最小生成树权值和 

 struct node
 {
     int x;
     int y;
 }PT[];//点信息 

 void Add_Edge(int u,int v,double w)
 {
     Edge[tot].to=v;
     Edge[tot].val=w;
     Edge[tot].next=Head[u];
     Head[u]=tot++;
 }

 double lowval[];
 int pre[];//记录每个点的双亲是谁

 double Prim(int n,int st)//n为顶点的个数,st为最小生成树的开始顶点
 {
     double sum=;
     fill(lowval,lowval+n,INF);//根据编号从0或是1开始，改+1
     memset(pre,-,sizeof(pre));
     lowval[st]=-;
     pre[st]=-;
     int num=;//num是目前可以添加到最小生成树中的点的数量
     for(int i=Head[st];i!=-;i=Edge[i].next)
     {
         int v=Edge[i].to;
         double w=Edge[i].val;
         lowval[v]=min(lowval[v],w);
         pre[v]=st;
         num++;
     }
     while(num)
     {
         double MIN=INF;
         int k;
         for(int i=;i<n;i++)//根据编号从0或是1开始，改i从0--n-1和1--n
         {
             if(lowval[i]!=-&&lowval[i]<MIN)
             {
                 MIN=lowval[i];
                 k=i;
             }
         }
         sum+=MIN;
         lowval[k]=-;
         num--;
         for(int j=Head[k];j!=-;j=Edge[j].next)
         {
             int v=Edge[j].to;
             double w=Edge[j].val;
             if(w<lowval[v])
             {
                 if(lowval[v]==INF)
                     num++;
                 lowval[v]=w;
                 pre[v]=k;
             }
         }
     }
     for(int i=;i<n;i++)//判断每个点的pre，看是否有双亲
     {
         if(i!=st&&pre[i]==-)
             return INF;
     }
     return sum;
 }

 int main()
 {
     int T;
     scanf("%d",&T);
     while(T--)
     {
         int n;
         scanf("%d",&n);
         memset(Head,-,sizeof(Head));
         tot=;
         for(int i=;i<n;i++)
             scanf("%d %d",&PT[i].x,&PT[i].y);
         for(int i=;i<n;i++)
         {
             for(int j=i+;j<n;j++)
             {
                 int x=PT[i].x-PT[j].x;
                 int y=PT[i].y-PT[j].y;
                 double val=sqrt(x*x+y*y);
                 if(val>=&&val<=)
                 {
                     Add_Edge(i,j,val);
                     Add_Edge(j,i,val);
                 }
             }
         }
         MST=Prim(n,);
         if(MST==INF)
             printf("oh!\n");
         else
             printf("%.1f\n",MST*);
     }
     return ;
 }