POJ - 2349 Arctic Network （最小生成树）

题目链接：点击打开链接

题意：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]);
	}
}