UVA10397 Connect the Campus（Prim最小生成树）

题意：给出n个点，再已知m个连接的边，求将所以点连接在一起的长度之和最小。

思路：Prim算法，因为m条边已知，所以在二维数组中将m条边的正反向长度设为0。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <string>
#include <algorithm>
#include <cmath>
#define INF 0x7fffffff

using namespace std;
double g[755][755],dis[755],x[755],y[755],ans;
int vis[755],n,m;

double prim(int x)
{
    vis[x]=1;
    for(int i=1;i<=n;i++)
    {
        if(vis[i]==0)
            dis[i]=g[x][i];
    }
    for(int i=2;i<=n;i++)
    {
        double mi=INF;
        int k;
        for(int j=1;j<=n;j++)
        {
            if(vis[j]==0&&mi>dis[j])
            {
                mi=dis[j];
                k=j;
            }
        }
        ans+=mi;
        vis[k]=1;
        for(int j=1;j<=n;j++)
        {
            if(vis[j]==0&&dis[j]>g[k][j])
                dis[j]=g[k][j];
        }
    }
    return ans;
}
int main()
{
    while(scanf("%d",&n)!=EOF)
    {
        ans=0;
        memset(vis,0,sizeof(vis));
        for(int i=1;i<=n;i++)
        {
            scanf("%lf%lf",&x[i],&y[i]);
        }
        for(int i=1;i<n;i++)
        {
            for(int j=i+1;j<=n;j++)
            {
                g[i][j]=g[j][i]=sqrt(pow((x[i]-x[j]),2)+pow((y[i]-y[j]),2));
            }
        }
        int u,v;
        scanf("%d",&m);
        while(m--)
        {
           scanf("%d%d",&u,&v);
           g[u][v]=g[v][u]=0;
        }
        printf("%.2f\n",prim(1));
    }
    return 0;

}