洛谷P1991 无线通讯网【最小生成树】

题目：https://www.luogu.org/problemnew/show/P1991

题意：有p个点的坐标，可以有s个点使得这s个点之间可以无限制通信。

要使所有点之间两两有通信的路径（可以是间接的），其中最长的一段通信路径是多长。

思路：最近都不好好看题。上一题没看清-1，这一题没看清精确到小数点两位，在干吗啊。

首先建图，然后找最小生成树里第p-s大的路径长度。

#include<cstdio>
#include<cstdlib>
#include<map>
#include<set>
#include<cstring>
#include<algorithm>
#include<vector>
#include<cmath> 
#include<stack>
#include<queue>
#include<iostream>

#define inf 0x3f3f3f3f
using namespace std;
typedef long long LL;
typedef pair<int, int> pr;

int s, n;
const int maxn = 505;
struct node{
    int x, y;
}nodes[maxn];
double g[maxn][maxn];

double dist(int i, int j)
{
    return sqrt((nodes[i].x - nodes[j].x) * (nodes[i].x - nodes[j].x) + (nodes[i].y - nodes[j].y) * (nodes[i].y - nodes[j].y));
}

bool vis[maxn];
double d[maxn];
vector<double>ans;
void prim(int x)
{
    memset(d, 0x3f, sizeof(d));
    d[x] = 0;
    vis[x] = true;
    for(int i = 1; i <= n; i++){
        d[i] = g[x][i];
    }
    
    for(int t = 1; t < n; t++){
        double min = inf;
        int minid;
        for(int i = 1; i <= n; i++){
            if(!vis[i] && d[i] < min){
                min = d[i];
                minid = i;
            }
        }
        vis[minid] = true;
        ans.push_back(min);
        for(int i = 1; i <= n; i++){
            if(d[i] > g[minid][i]){
                d[i] = g[minid][i];
            }
        }
    }
}

int main()
{
    scanf("%d%d", &s, &n);
    for(int i = 1; i <= n; i++){
        scanf("%d%d", &nodes[i].x, &nodes[i].y);
    }
    for(int i = 1; i <= n; i++){
        for(int j = i + 1; j <= n; j++){
            g[i][j] = g[j][i] = dist(i, j);
            //printf("%lf
", g[i][j]);
        }
    }
    
    prim(1);
    sort(ans.begin(), ans.end());
    printf("%.2lf
", ans[n - s - 1]);
    return 0;
}