2019_GDUT_新生专题 图论 --- E

E — 最短路

原题链接：https://vjudge.net/contest/351234#problem/E

题目大意：

有多个哨站，每个哨站都有一部视频联络机，一些哨站还同时配有卫星联络。配有卫星联络的哨站可不限距离相互联系，而只有视频联络机的只能在一定范围内联系（限制距离），距离越大，花费越大。给出哨站的数目、坐标和配有卫星联络的数目，求最小的限制距离。

题目分析：

这是最小生成树问题，一般用Kruskal算法、Prim算法。
使用Kruskal算法。从最小边开始，一步步构建树，当刚好生成树时，就是最小生成树。题目的只需在构建过程中剩余s个点，就可以退出构建，输出答案（引起退出的那条边）。

代码实现：

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cmath>
//Kruskal算法
using namespace std;

int x[501], y[501];//坐标
int d[501];//各点关系
struct edge
{
    int u;
    int v;
    double len;
};
//储存两点距离
edge e[125000];
//找出x的代表根
int find_r(int x)
{
    int r = x;
    while(d[r] != r) r = d[r];
    return r;
}
//x的平方
int square(int x)
{
    return x * x;
}
//依据距离比较，从小到大
bool cmp(edge x, edge y)
{
    if(x.len < y.len) return true;
    else return false;
}

int main()
{
    int n;
    int s, p, num;
    double dist, ans;
    int fa, fb;
    int i, j, k;

    scanf("%d",&n);
    while(n--)
    {
        scanf("%d%d",&s,&p);
        //初始化，各点相互独立
        for(i=1; i<=p; i++){
            d[i] = i;
        }

        for(i=1; i<=p; i++){
            scanf("%d%d",&x[i],&y[i]);
        }

        //计算每两个点的距离
        k = 1;
        for(i=1; i<=p; i++){
            for(j=i+1; j<=p; j++){
                e[k].u = i;
                e[k].v = j;
                dist = square(abs(x[i] - x[j])) + square(abs(y[i] - y[j]));
                e[k].len = sqrt(dist);
                k++;
            }
        }
        k--;

        //排序
        sort(e+1,e+k+1,cmp);

        //建立联系，即最小生成树
        num = 0; //建立联系的个数-1
        for(i=1; i<=k; i++){
            fa = find_r(e[i].u);
            fb = find_r(e[i].v);
            if(fa != fb){
                d[fa] = fb;
                num++;
                if(num >= p - s){ //只需要p-s+1个用视频，其他的靠卫星
                    ans = e[i].len;
                    break;
                }
            }
        }
        printf("%.2f\n",ans);
    }
    return 0;
}

最后，希望路过的dl们能给予改进的建议！