哈理工OJ 1630 网线 【建图+Prime算法】

题目链接：http://acm.hrbust.edu.cn/index.php?m=ProblemSet&a=showProblem&problem_id=1630

网线
Time Limit: 5000 MS Memory Limit: 32768 K
Total Submit: 82(22 users) Total Accepted: 23(20 users) Rating: Special Judge: No
Description
试设计一个网络连接某个区域中的一些地点。给定这些地点所在的位置，网线的长度就是两个地点的线段距离，假定，给定的网线可以直接或间接的连接该地区中的所有点，试为这个地区设计一个网络系统，使得该地区所有地点都可以直接或间接的连接，并且使用网线长度最短。

Input
输入一个数字n（n<1000），给出这n个点的位置（x,y），即坐标（0≤x＜1000，0≤y＜1000）。

Output
输出网线的最短距离，保留小数点后四位。
Sample Input
1

1 1

2

0 0

1 1

3

0 0

0 1

1 1

Sample Output
0.0000

1.4142

2.0000

【分析】用Kruskal算法会超时，单用Prime算法可以直接过。

下面是 AC代码：

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

const double INF=1000000;
const int MAXN=1100;
bool vis[MAXN];
double lowc[MAXN];
double a[MAXN][MAXN];

double Prime(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;
}

struct Point
{
    double x,y;
}p[1005];

double dist(Point p1,Point p2)
{
    return sqrt((p1.x-p2.x)*(p1.x-p2.x)+(p1.y-p2.y)*(p1.y-p2.y));
}

int main()
{
    int n;
    while(~scanf("%d",&n))
    {
        for(int i=0;i<n;i++)
        {
            scanf("%lf%lf",&p[i].x,&p[i].y);
        }
        memset(a,INF,sizeof(a));
        for(int i=0;i<n;i++)//建图
        {
            for(int j=i+1;j<n;j++)
            {
                   a[i][j]=dist(p[i],p[j]);
                   a[j][i]=a[i][j];
            }
        }
        double re=0;
        re=Prime(a,n);
        printf("%.4lf\n",re);
    }
    return 0;
}