poj 2349 最小生成树

点击打开链接
#include <iostream>
#include<cstdio>
#include<algorithm>
#include<cmath>
using namespace std;
typedef pair<int,int> PII;
const int maxv=500+5;
const int maxe=250000+5;
struct Edge{
int from,to,weight;
};
bool operator<(const Edge &E1,const Edge &E2)
{
    return E1.weight<E2.weight;
}
int parent[maxv];
PII v[maxv];
Edge edges[maxe];
int find(int x)
{
    return x==parent[x]?x:parent[x]=find(parent[x]);
}
int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        int s,p;        //卫星数量，顶点数量
        scanf("%d%d",&s,&p);
        for(int i=1;i<=p;i++)
            parent[i]=i;
        for(int i=0;i<p;i++)
        {
            scanf("%d%d",&v[i].first,&v[i].second);
        }
        int cnt=0;
        for(int i=0;i<p-1;i++)
        {
            for(int j=i+1;j<p;j++)
            {
                edges[cnt].from=i+1;
                edges[cnt].to=j+1;
                edges[cnt].weight=(v[i].first-v[j].first)*(v[i].first-v[j].first)+(v[i].second-v[j].second)*(v[i].second-v[j].second);
                cnt++;
            }
        }
        sort(edges,edges+cnt);
        int f=p-s;
        for(int i=0;i<cnt;i++)
        {
            int p1=find(edges[i].from);
            int p2=find(edges[i].to);
            if(p1==p2) continue;
            parent[p1]=p2;
            f--;
            if(f==0) {
                printf("%.2f\n",sqrt(edges[i].weight));
                break;
            }
        }
    }
    return 0;
}