题意:
太空帝国要通过建造隧道来联通它的 N 个星球。
每个星球用三维坐标(xi,yi,zi) 来表示,而在两个星球 A,B之间建造隧道的价格为
min{∣xA−xB∣,∣yA−yB∣,∣zA−zB∣}。
现要建造 N-1条隧道使得所有的星球都能直接或间接相连。求完成该任务所需的最小总价。
思路:
首先把题意理解清楚就是,对n个点要建n-1条边然后完成全部相连,这不一最小生成树吗,脑子里就两个算法,克鲁斯卡尔和普罗米。
Prim 算法
n^2的复杂度,实现起来就是在一个无向图,在图中选择若干条边把图的所有节点连起来。要求边长之和最小。在图论中,叫做求最小生成树。
而实现起来就像是迪杰斯特拉,但是有不一样的地方。
注意上述两个伪算法的差别只在于最后循环体内的松弛操作。
- 最小生成树只关心所有边的和最小,所以有v.key = w(u, v),即每个点直连其他点的最小值(最多只有两个节点之间的权值和)
- 最短路径树只搜索权值最小,所以有v.key = w(u, v) + u.key,即每个点到其他点的最小值(最少是两个节之间的权值和)
然后复杂度大概是
Kruskal算法
方法:并查集,每次将两个集合找最小边合并找最小值即可。
时间复杂度: O(nlogn)看上去是比较可行的,1e5不会t!盘他
接下来思考如何建边,再观察边权,两个点之间的边权居然是它们同系坐标里的最小值,那如果每个点都去暴力的求三维的边理应会有条边,显然直接爆炸了,不可行,因为每次两个点相连其实只用到了一条边,所以思考如何优化?
所以我们可以以三维来排序,一共排三次,每次排完都将相邻的两个点连一条边。如果我们使用了不符合题设的边作为答案,那么它一定可以被符合题设的边替代。
那么为啥排序后相邻的两个点一定最优呢?其实不难想用贪心的思考方式,排序之后假设只看x轴
当然x1--->x2---->x3,如果需要连他们肯定是连x1和x2或者x2和x3,而不是选择x1和x3,这样边就可以省略了,每个坐标系都这么做,就只需要建最多3*n条边即可!
接下来就直接上克鲁斯卡尔的板子就行,需要注意的是这里有3*n条边,当n-1条边完成时,最小生成树已经建立完毕,直接跳出。
#include<bits/stdc++.h>
using namespace std;
const int maxn=3e5+100;
int tot,n;
struct node {
int id;
long long len;
bool operator<(const node &a){
return len<a.len;
}
}x[maxn],y[maxn],z[maxn];
struct edge
{
long long dis;
int u,v;
bool operator < (const edge &x)const
{
return dis<x.dis;
}
}e[5*maxn];
void add(int i,int j)
{
e[++tot].dis=abs(x[i].len-x[j].len);
e[tot].u=x[i].id; e[tot].v=x[j].id;
e[++tot].dis=abs(y[i].len-y[j].len);
e[tot].u=y[i].id; e[tot].v=y[j].id;
e[++tot].dis=abs(z[i].len-z[j].len);
e[tot].u=z[i].id; e[tot].v=z[j].id;
}
int f[100010];
int findx(int x)
{
if(x==f[x]) return x;
else return f[x]=findx(f[x]);
}
void kruskal()
{
long long res=0,cnt=0;
sort(e+1,e+tot+1);//排序所有边
for(int i=0;i<=n;i++) f[i]=i;//初始化并查集
for(int i=1;i<=tot;i++)//从小到大枚举所有边
{
int a=e[i].u,b=e[i].v;
long long w=e[i].dis;
a=findx(a),b=findx(b);
// cout<<a<<" "<<b<<endl;
if(a!=b) //如果a和b不连通,集合里加入这条边
{
res+=w;//最小生成树中所有边的和
f[a]=b;
cnt++;
if(cnt==n-1) break;
}
// cout<<res<<" "<<cnt<<" "<<n-1<<endl;
}
cout<<res<<endl;
}
int main()
{
int i,j;
cin>>n;
for(i=0;i<n;i++)
{
cin>>x[i].len>>y[i].len>>z[i].len;
x[i].id=i;y[i].id=i;z[i].id=i;
}
sort(x,x+n);sort(y,y+n);sort(z,z+n);
for(i=0;i<n-1;i++)
{
add(i,i+1);
}
kruskal();
// for(i=1;i<=tot;i++)
// {
// cout<<f[i]<<endl;
// }
return 0;
}