HDU1875+Prim模板

https://cn.vjudge.net/problem/HDU-1875

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

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

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

2
2
10 10
20 20
3
1 1
2 2
1000 1000

Sample Output

1414.2
oh!

 #include<iostream>
 #include<stdio.h>
 #include<bits/stdc++.h>
 const int inf=;
 using namespace std;
 int x[],y[];
 double w[][],dis[];
 int n,m;
 bool vis[];
 double weight(int i,int j)
 {
     return sqrt(1.0*((x[i]-x[j])*(x[i]-x[j])+(y[i]-y[j])*(y[i]-y[j])));
 }
 void prim()
 {
     int i,j,k;
     bool flag=;
     double sum=,tmp;
     for(int i=;i<=m;i++)
         dis[i]=inf;
     dis[]=;
     for(i=;i<=m;i++)
     {
         tmp=inf;
         for(j=;j<=m;j++)
          if(!vis[j]&&tmp>dis[j])
          {
              tmp=dis[j];
              k=j;
          }
          if(tmp==inf){flag=;break;}
          vis[k]=;
          sum+=tmp;
          int y;
          for(j=;j<=m;j++)
          {
              if(!vis[j]&&dis[j]>w[k][j])
                 dis[j]=w[k][j];
          }

     }
     if(flag)
         printf("%.1f\n",sum*);
     else
         printf("oh!\n");
 }

 int main()
 {
     cin>>n;
     for(int i=;i<n;i++)
     {
         cin>>m;
         memset(x,,sizeof x);
         memset(y,,sizeof y);
         memset(vis,,sizeof vis);
         for(int j=;j<=m;j++)
         cin>>x[j]>>y[j];
         for(int k=;k<=m;k++)
             for(int l=;l<=m;l++)
         {
             double p=weight(k,l);
             if(p>=&&p<=)
                 w[k][l]=w[l][k]=p;
             else
                 w[k][l]=w[l][k]=inf;
         }
         prim();
     }
     return ;
 }

思路：

使用Prim求解最小生成树。

注意点：

1.一定要记得给每个数组初始化，一开始忘记给vis[]初始化了，导致调试了好久。

2.用memset的时候要注意类型，我给double赋极大值错误地使用了memset(dis,0x3f,sizeof dis);后来想到double类型是8个字节的，int是四个字节的（这是四字节赋极大值的方法，赋给8个字节的double类型的数得到的结果不一样），因为memset是按字节赋值的。这么给double赋值得到了一个约等于0的数，与预想的结果不同，导致错误。

Prim模板（解决最小生成树问题）：

 void prim()
 {
     int i,j,k,tmp;
     memset(dis,0x3f,sizeof dis);
     dis[]=;
     for(i=;i<=n;i++)
     {
         tmp=inf;
         for(j=;j<=n;j++)
          if(!vis[j]&&tmp>dis[j])
          {
              tmp=dis[j];
              k=j;
          }//找到最小距离的节点
          vis[k]=;
          for(j=;j<=n;j++)
          {
              if(!vis[j]&&dis[j]>g[k][j])
                 dis[j]=g[k][j];
          }//更新最短距离
     }
 }