hdu 1875 畅通工程再续(最小生成树)

思路：最小生成树模板题。

#include<iostream>
#include<stdio.h>
#include<string.h>
#include<math.h>
using namespace std;

const int INF=0x3f3f3f3f;
const int MAXN=110;
bool vis[MAXN];
double lowc[MAXN];

double Prim(double cost[][MAXN],int n)
{
    double ans=0;
    memset(vis,false,sizeof(vis));
    vis[0]=true;
    for(int i=1; i<n; i++)lowc[i]=cost[0][i];
    for(int i=1; i<n; i++)
    {
        double minc=INF;
        int p=-1;
        for(int j=0; j<n; j++)
            if(!vis[j]&&minc>lowc[j])
            {
                minc=lowc[j];
                p=j;
            }
        if(minc==INF)return -1;//原图不连通
        ans+=minc;
        vis[p]=true;
        for(int j=0; j<n; j++)
            if(!vis[j]&&lowc[j]>cost[p][j])
                lowc[j]=cost[p][j];
    }
    return ans;
}

double cost[MAXN][MAXN];

int p[128][2];
int tol;

void make()
{
    for(int i=0; i<tol; ++i)
    {
        for(int j=0; j<tol; ++j)
        {
            cost[i][j]=sqrt((p[i][0]-p[j][0])*(p[i][0]-p[j][0])+
                            (p[i][1]-p[j][1])*(p[i][1]-p[j][1]));
            if(cost[i][j]<10||cost[i][j]>1000)
                cost[i][j]=INF;
        }
    }
}

int main()
{
    int T;
    int C;

    scanf("%d",&T);
    while(T--)
    {
        tol=0;
        scanf("%d",&C);
        for(int i=0; i<C; ++i)
        {
            scanf("%d%d",&p[tol][0],&p[tol][1]);
            ++tol;
        }
        make();

        double ans=Prim(cost,C);

        if(ans==-1)printf("oh!\n");
        else printf("%.1f\n",ans*100);
    }
    return 0;
}