题目描述
某国有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
题解:
5000 * 5000的空间大小会MLE,那么我们发现题目给的是坐标,所以我们想到可以不直接存储距离,而是在Prim的运行流程中把距离即时计算出来即可。
AC代码
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cmath>
#define INF 0x3f3f3f3f
using namespace std;
struct position{
int x,y;
}room[5001];
bool vis[5001];
double low[5001];
int n;
double len(int i,int j){
return sqrt((double)(room[i].x-room[j].x)*(room[i].x-room[j].x)+(double)(room[i].y-room[j].y)*(room[i].y-room[j].y));
}
void prim(){
for(int i=1;i<=n;i++){
low[i]=len(1,i);
//cout<<low[i]<<' ';
}
low[1]=0;vis[1]=true;
for(int i=1;i<=n;i++){
int minn=INF;
int temp;
for(int j=1;j<=n;j++){
//cout<<vis[j]<<' '<<low[j]<<"--"<<endl;
if(!vis[j]&&minn>low[j]){
minn=low[j];
temp=j;
}
}
//cout<<minn<<"--"<<endl;
vis[temp]=true;
for(int j=1;j<=n;j++){
if(!vis[j]&&low[j]>len(temp,j)){
low[j]=len(temp,j);
}
}
}
}
int main(){
scanf("%d",&n);
//cout<<vis[1]<<endl;
for(int i=1;i<=n;i++){
scanf("%d%d",&room[i].x,&room[i].y);
}
prim();
double ans=0;
for(int i=1;i<=n;i++){
ans+=low[i];
//cout<<low[i]<<endl;
}
printf("%.2lf\n",ans);
}