UVA - 10397 - Connect the Campus

题目链接

题意是要你用最短的缆绳连接校园里的所有建筑，其本质是一个最小生成树的问题。题目给出的已经安装好的缆绳权值赋零，然后用邻接矩阵存储图，计算各点直接的距离，再用Prim算法求出最小生成树上的所有权值之和即可。

#include<stdio.h>
#include<string.h>
#include<math.h>
#define MAX 2147483647
typedef struct no
{
    int x,y;
}Node;
Node build[800];
double graph[800][800],lowcost[800];   
int vis[800]; 
double dis(int x1,int y1,int x2,int y2)
{
    return sqrt((x1-x2)*(x1-x2)+(y1-y2)*(y1-y2));
}    
void AcquireGraph(int n)
{
    int i,j;
    for(i=1;i<=n;i++)
    {
        for(j=1;j<=n;j++)
        {
            if(i==j)continue;
            graph[i][j]=dis(build[i].x,build[i].y,build[j].x,build[j].y);
        }
    }
}            
double Prim(int n)
{
    int i,j,v;
    double min,sum=0;
    memset(vis,0,sizeof(vis));
    for(i=1;i<=n;i++)
    {
        lowcost[i]=graph[1][i];
    }
    vis[1]=1;
    for(i=1;i<=n;i++)
    {
        min=MAX;  
        v=0;
        for(j=1;j<=n;j++)
        {
            if(lowcost[j]<min&&!vis[j])
            {
                min=lowcost[j];  
                v=j;
            }   
        }    
        if(v)
        {
            vis[v]=1;
            sum+=min;
            for(j=1;j<=n;j++)
            {
                if(!vis[j]&&lowcost[j]>graph[v][j])
                {
                    lowcost[j]=graph[v][j];
                }  
            }
        }   
    } 
    return sum;                 
}                                   
int main()
{
    int N,M;
    int i,j,k,u,v;
    double sum;
    while(scanf("%d",&N)!=EOF)
    {
        for(i=1;i<=N;i++)
        {
            for(j=1;j<=N;j++)
            {
                if(i==j)graph[i][j]=0;
                else graph[i][j]=MAX;
            }
        }        
        for(i=1;i<=N;i++)
        {
            scanf("%d%d",&build[i].x,&build[i].y);
        }    
        AcquireGraph(N);
        scanf("%d",&M);
        for(i=0;i<M;i++)
        {
            scanf("%d%d",&u,&v);
            graph[u][v]=graph[v][u]=0;
        }    
        sum=Prim(N);
        printf("%.2lf\n",sum);
    }    
    return 0;
}