【YBTOJ进阶训练指导】最小距离和【最小生成树】【贪心】

思路：

我们发现只有距离最近的两点才会被连边，于是我们按照x,y,z分别排一次序，然后统计出可能连的边，然后做最小生成树

$code$

#include<iostream>
#include<cstdio>
#include<algorithm>

using namespace std;

int n, tot, num;
int fa[101000];
struct node
{
	int x, y, z, id;
}a[101000];
struct abc
{
	int x, y, w;
}b[501000];

bool cmp1(node x, node y)
{
	return x.x<y.x;
}

bool cmp2(node x, node y)
{
	return x.y<y.y;
}

bool cmp3(node x, node y)
{
	return x.z<y.z;
}

bool cmp(abc x, abc y)
{
	return x.w<y.w;
}

int find_(int x)
{
	if(fa[x]==x)
		return x;
	return fa[x]=find_(fa[x]);
}

int main()
{
	scanf("%d", &n);
	for(int i=1; i<=n; i++)
		scanf("%d%d%d", &a[i].x, &a[i].y, &a[i].z), a[i].id=i, fa[i]=i;
	sort(a+1, a+1+n, cmp1);
	for(int i=1; i<n; i++)
		b[++tot].x=a[i].id, b[tot].y=a[i+1].id, b[tot].w=a[i+1].x-a[i].x;
	sort(a+1, a+1+n, cmp2);
	for(int i=1; i<n; i++)
		b[++tot].x=a[i].id, b[tot].y=a[i+1].id, b[tot].w=a[i+1].y-a[i].y;
	sort(a+1, a+1+n, cmp3);
	for(int i=1; i<n; i++)
		b[++tot].x=a[i].id, b[tot].y=a[i+1].id, b[tot].w=a[i+1].z-a[i].z;
	sort(b+1, b+1+tot, cmp);
	int ans=0;
	for(int i=1; i<=tot; i++)
	{
		int fx=find_(b[i].x), fy=find_(b[i].y);
		if(fx!=fy)
		{
			fa[fy]=fx;
			ans+=b[i].w;
			num++;
			if(num==n-1)
				break;
		}
	}
	printf("%d", ans);
	return 0;
}