HDU - 1875 畅通工程再续 prim算法

/**
这个题目就是在原来畅通工程的基础上加了一个条件，说白了看是否能够联通；
如果能够联通，算出最低花费；没有什么难度，水水的就过了；
*/

#include<cstdio>
#include<cmath>
using namespace std;

int n;
const double inf=0x3f3f3f3f*1.0;
const int maxn=2e2+7;
double mmp[maxn][maxn],ans,dis[maxn];
bool vis[maxn];

struct node
{
    int x,y;
}edge[maxn];

double far(node a,node b)
{
    return sqrt((double)((a.x-b.x)*(a.x-b.x)+(a.y-b.y)*(a.y-b.y)));
}

void build_map()
{
    for(int i=1;i<n;i++)
        for(int j=i+1;j<=n;j++)
        {
            double len=far(edge[i],edge[j]);
            if(len<10||len>1000)
            {
                mmp[i][j]=inf;
                mmp[j][i]=inf;
            }
            else
            {
                len*=100;
                mmp[i][j]=len;
                mmp[j][i]=len;
            }
        }
}

void prim()
{
    ans=0;
    for(int i=1;i<=n;i++)
    {
        vis[i]=0;
        dis[i]=mmp[1][i];
    }
    vis[1]=1;
    int flag=1;
    for(int i=1;i<n;i++)
    {
        double Min=inf;
        int k=-1;
        for(int j=2;j<=n;j++)
        {
            if(!vis[j]&&dis[j]<Min)
            {
                Min=dis[j];
                k=j;
            }
        }
        if(k==-1){printf("oh!\n");return ;}//存在未被访问的点，如果满足条件的话，不会中途停止的；
        ans+=Min;
        vis[k]=1;
        for(int j=2;j<=n;j++)
            if(!vis[j]&&dis[j]>mmp[k][j])
            dis[j]=mmp[k][j];
    }
    printf("%.1f\n",ans);
}

int main ()
{
    int t;
    scanf("%d",&t);
    while(t--)
    {
      scanf("%d",&n);

      for(int i=1;i<=n;i++)
        for(int j=1;j<=n;j++)
        if(i==j) mmp[i][j]=0;
      else mmp[i][j]=inf;

      for(int i=1;i<=n;i++)
        scanf("%d %d",&edge[i].x,&edge[i].y);
        build_map();
        prim();
    }
    return 0;
}