洛谷 P1265 公路修建 ( 最小生成树prim) 题解

题目来源:

https://www.luogu.org/problemnew/show/P1265

题目描述:

某国有n个城市,它们互相之间没有公路相通,因此交通十分不便。为解决这一“行路难”的问题,政府决定修建公路。修建公路的任务由各城市共同完成。

修建工程分若干轮完成。在每一轮中,每个城市选择一个与它最近的城市,申请修建通往该城市的公路。政府负责审批这些申请以决定是否同意修建。

政府审批的规则如下:

(1)如果两个或以上城市申请修建同一条公路,则让它们共同修建;

(2)如果三个或以上的城市申请修建的公路成环。如下图,A申请修建公路AB,B申请修建公路BC,C申请修建公路CA。则政府将否决其中最短的一条公路的修建申请;

(3)其他情况的申请一律同意。

一轮修建结束后,可能会有若干城市可以通过公路直接或间接相连。这些可以互相:连通的城市即组成“城市联盟”。在下一轮修建中,每个“城市联盟”将被看作一个城市,发挥一个城市的作用。

当所有城市被组合成一个“城市联盟”时,修建工程也就完成了。

你的任务是根据城市的分布和前面讲到的规则,计算出将要修建的公路总长度。

输入输出格式

输入格式:

第一行一个整数n,表示城市的数量。(n≤5000)

以下n行,每行两个整数x和y,表示一个城市的坐标。(-1000000≤x,y≤1000000)

输出格式:

一个实数,四舍五入保留两位小数,表示公路总长。(保证有惟一解)

输入输出样例

输入样例#1: 复制

4
0 0
1 2
-1 2
0 4

输出样例#1: 复制

6.47

说明

修建的公路如图所示: 

解题思路:

      第二个条件显然不成立,画一个图就可以判断,所以其实题目就是求最小生成树,因为点边很多,我们只能用prim来求最小生成树,再求的时候计算距离。。。

代码:

#include <iostream>
#include <algorithm>
#include <cstring>
#include <string>
#include <vector>
#include <cmath>
using namespace std;
int father[5005];
int n,m;
double dis[5005];
struct newtt{
    double x,y;
}bian[5005];
double js(newtt a,newtt b)
{
    return sqrt((a.x-b.x)*(a.x-b.x)+(a.y-b.y)*(a.y-b.y));
}
bool vis[5005];
int main()
{
    cin>>n;
    memset(vis,0,sizeof(vis));
    for(int i=1;i<=n;i++)
    {
        scanf("%lf%lf",&bian[i].x,&bian[i].y);
        dis[i]=1e10;
    }
    
    for(int i=1;i<=n;i++)
    {
        dis[i]=js(bian[1],bian[i]);
    }
    vis[1]=1;
    dis[1]=0;
    double ans=0;
    for(int i=1;i<=n;i++)
    {
        double maxn=1e10;
        int t=-1;
        for(int j=1;j<=n;j++)
        {
            if(!vis[j]&&maxn>dis[j]){
                t=j,maxn=dis[j];
            }
        }
        if(t!=-1){
            vis[t]=1,ans+=maxn;
            for(int j=1;j<=n;j++)
            {
                double kk=js(bian[t],bian[j]);
                if(dis[j]>kk)dis[j]=kk;
            }
        }
        
    }
    printf("%.2lf",ans);
    return 0;
}

猜你喜欢

转载自blog.csdn.net/qq_40400202/article/details/82696083
今日推荐