ACdream 孟竹要减肥？（prim）

刚开始题目看错了，一直wa，输入的坐标是浮点型。。。。。。。

听了队友的，才知道prim算法只需要枚举到sqrt（n）就完全足够了，发现sqrt（n）好多地方用的到。

#include<stdio.h>
#include<math.h>
 
const int MAXN=510;
const int INF=0x7fffffff;
struct Node
{
    double x,y;
} node[MAXN];
 
double temp;
double dist[MAXN];//dist[i]表示i向外延伸的最短边长
double map[MAXN][MAXN];//储存a->b之间的边权值
int pre[MAXN];//pre[i]记录i的父节点
int n,m;
 
double _dist(Node a,Node b)
{
    return sqrt((a.x-b.x)*(a.x-b.x)+(a.y-b.y)*(a.y-b.y));
}
 
void Prim(int s)
{
    int i,j,k;
    double min;
    bool p[MAXN];//区分Va集合和Vb集合
    for(i=1; i<=n; i++)
    {
        p[i]=false;
        dist[i]=map[s][i];
        pre[i]=s;
    }
    dist[s]=0;
    p[s]=true;
    for(i=1; i<=n-1; i++) //循环n-1次，每次加入一个点
    {
        min=INF;
        k=0;
        for(j=1; j<=n; j++)
        {
            //若j还在Vb集合中，并且j的路径可以再小点(找当前i点所对应的最短权值)
            if(!p[j] && dist[j]<min)
            {
                min=dist[j];
                k=j;
            }
        }
        if(k==0) return ;//如果没有点可以扩展，即图G不连通，返回
        if(temp<dist[k]) temp=dist[k];
        if(i==m) return ;
        p[k]=true;//将点k移到Va集合
        for(j=1; j<=n; j++) //更新相邻k相邻的点（有些dist依旧记录着以前Va结点中相邻的点）
        {
            //对于每个与k相邻的Vb中的点j，更新j距离最近的点及其距离
            if(!p[j] && map[k][j]!=INF && dist[j]>map[k][j])
            {
                dist[j]=map[k][j];
                pre[j]=k;
            }
        }
    }
}
 
int main()
{
    int T,i,j;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d%d",&n,&m);
        for(i=1; i<=n; i++)
        {
            scanf("%lf%lf",&node[i].x,&node[i].y);
        }
        for(i=1; i<=n; i++)
        {
            for(j=1; j<=n; j++)
            {
                map[i][j]=_dist(node[i],node[j]);
            }
        }
        double res=INF;
        for(i=1; i*i<=n; i++)//////
        {
            temp=0;
            Prim(i);
            if(res>temp) res=temp;
        }
        printf("%.4lf\n",res);
    }
    return 0;
}