Treehouses

题目链接
用到了最小生成树，先将p个已经用缆线连接的两点间距离设为0，再计算每个点之间的距离，如果是前e个输入的，则其两点间的距离为0，否则计算其距离。再将两点间的距离升序排序，然后就是用最小生成树的思路去做。

借鉴VJ账号为Get_rid_of_WA大佬的解法

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cmath>
#include <algorithm>
#include <string>
#include <cstring>
#include <vector>
#include <map>
#include <set>
#include <queue>
#include <unordered_map>
#define ll long long
using namespace std;
struct node 
{
    double x,y;
}point[1005];
struct Edge
{
    int u,v;
    double dis;
}temp;
int n,p,e,u,v;
int f[1005];
vector<Edge>edg;

void init()
{
    for(int i=1; i<=n; i++)
        f[i] = i;
}

int find(int x)
{
    if(x!=f[x])
        f[x] = find(f[x]);
    return f[x];
}

void join(int x,int y)
{
    x = find(x);
    y = find(y);
    if(x!=y)
        f[y] = x;
}

bool cmp(Edge a,Edge b)
{
    return a.dis<b.dis;
}

bool same(int x,int y)
{
    x=find(x);
    y=find(y);
    if(x==y)
        return true;
    else
        return false;
}

double solve()
{
    sort(edg.begin(),edg.end(),cmp);
    init();
    double res=0;
    for(int i=0; i<edg.size(); i++)
    {
        temp = edg[i];
        if(!same(temp.u,temp.v))
        {
            join(temp.u,temp.v);
            res+=temp.dis;
        }
    }
    return res;
}

int main()
{
    scanf("%d%d%d",&n,&e,&p);
    for(int i=1; i<=n; i++)
    {
        scanf("%lf%lf",&point[i].x,&point[i].y);
    }
    while(p--)//已经用缆绳连接，dis=0
    {
        cin>>u>>v;
        temp.u = u;
        temp.v = v;
        temp.dis=0;
        edg.push_back(temp);
    }
    for(int i=1; i<n; i++)
    {
        for(int j=i+1; j<=n; j++)
        {
            if(i<=e&&j<=e)//如果小于e，则该点靠近空地
            {
                temp.u=i;
                temp.v=j;
                temp.dis=0;
                edg.push_back(temp);
            }
            else
            {
                temp.u=i;
                temp.v=j;
                temp.dis = sqrt(pow(abs(point[i].x-point[j].x),2)+pow(abs(point[i].y-point[j].y),2));
                edg.push_back(temp);
            }
        }
    }
    double ans = solve();
    printf("%.7lf\n",ans);
    return 0;
}