哈理工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;}