题目链接:点击打开链接
题意:n个村子要完成通信,有两种方式可以让村子之间通信,第一种是拉网线,花费为存在之间的距离,第二种为按卫星,若两个村子都按了卫星那么他们就可以通信。
1-----2 ------3
比如1和2直接连了网线 2和3有网线, 1 和 3也是能够通信的。
题解:m个卫星就代表可以少建m-1条边。然后kruskal跑的时候只建(n-1)-(m-1)条边就解决了。
代码如下
#include<cmath>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
#define inf 0x3f3f3f3f
int f[505];
double num[505];
struct node{
int x;
int y;
}s[505];
struct edge{
int x;int y;
double cost;
void Edge(int x1,int y1,double W){
x = x1;
y = y1;
cost = W;
}
}e[250050];
double dis(node a,node b){
return sqrt(double((a.x - b.x) *(a.x - b.x) + (a.y - b.y) * (a.y -b.y)));
}
bool cmp(edge a,edge b){
return a.cost < b.cost;
}
int getf(int v){
if(f[v] != v){
f[v] = getf(f[v]);
}
return f[v];
}
void init(int n){
for(int i = 1; i <= n ; i ++)
f[i] = i;
}
int main(){
int z;
int S,P;
scanf("%d",&z);
while(z--){
int step = 0 ;
scanf("%d%d",&S,&P);
init(P);
for(int i = 1; i <= P ;i ++){
scanf("%d%d",&s[i].x,&s[i].y);
}
for(int i = 1; i <= P ; i ++){
for(int j = 1; j <=P ; j ++){
e[step++].Edge(i,j,dis(s[i],s[j]));
}
}
sort(e,e+step,cmp);
int p = 0 ;
for(int i = 0; i < step ; i ++){
int t1 = getf(e[i].x);
int t2 = getf(e[i].y);
if(t1 != t2){
f[t1] = t2;
num[p++] = e[i].cost;
}
}
printf("%.2f\n",num[P-S-1]);
}
}