J - Building a Space Station (最小生成树&kruskal)

思路:将所有两两的点的距离进行排序,然后用kruskal板子就行,唯一需要注意的是距离为负数不用加。

AC代码:ps(C++ac,G++wa)

#include<cstdio>
#include<cstring>
#include<iostream>
#include<queue> 
#include<cmath>
#include<algorithm>
#include<vector>
using namespace std;
const int N=205;
struct p{
	double x,y,z,r;
}a[N];
struct node{
	int u,v;
	double w;
	bool operator<(const node&a)const{
		return w<a.w;
	}
}e[N*N];
int s[N];
int find(int x){
	if(s[x]!=x) s[x]=find(s[x]);
	return s[x];
}
int main(){
	int n;
	while(~scanf("%d",&n)&&n){
		int k=0;
		for(int i=1;i<=n;i++) s[i]=i;
	for(int i=1;i<=n;i++){
		scanf("%lf%lf%lf%lf",&a[i].x,&a[i].y,&a[i].z,&a[i].r);
		for(int j=1;j<i;j++)
		{
			double dis=((a[i].x-a[j].x)*(a[i].x-a[j].x)+(a[i].y-a[j].y)*(a[i].y-a[j].y)+(a[i].z-a[j].z)*(a[i].z-a[j].z));
			dis=sqrt(dis),dis-=(a[i].r+a[j].r);
			e[++k].u=i,e[k].v=j,e[k].w=dis;
		}
	}
	sort(e+1,e+1+k);
	int tot=0;
	double ans=0;
		for(int i=1;i<=k;i++){
			int fa=find(e[i].u),fb=find(e[i].v);
			if(fa==fb) continue;
			tot++;
			s[fb]=fa;
			if(e[i].w>0) ans+=e[i].w;
			if(tot==n-1) break;
		}
		printf("%.3lf\n",ans);
	}
	return 0;
}