5.3/4每日一题——SVEMIR

题意:

太空帝国要通过建造隧道来联通它的 N 个星球。

每个星球用三维坐标(xi​,yi​,zi​) 来表示,而在两个星球 A,B之间建造隧道的价格为

min{∣xA​−xB​∣,∣yA​−yB​∣,∣zA​−zB​∣}。

现要建造 N-1条隧道使得所有的星球都能直接或间接相连。求完成该任务所需的最小总价。

思路:

首先把题意理解清楚就是,对n个点要建n-1条边然后完成全部相连,这不一最小生成树吗,脑子里就两个算法,克鲁斯卡尔和普罗米。

Prim 算法

n^2的复杂度,实现起来就是在一个无向图,在图中选择若干条边把图的所有节点连起来。要求边长之和最小。在图论中,叫做求最小生成树。

而实现起来就像是迪杰斯特拉,但是有不一样的地方。

注意上述两个伪算法的差别只在于最后循环体内的松弛操作。

  1. 最小生成树只关心所有边的和最小,所以有v.key = w(u, v),即每个点直连其他点的最小值(最多只有两个节点之间的权值和)
  2. 最短路径树只搜索权值最小,所以有v.key = w(u, v) + u.key,即每个点到其他点的最小值(最少是两个节之间的权值和)

然后复杂度大概是n^{2}

Kruskal算法

方法:并查集,每次将两个集合找最小边合并找最小值即可。

时间复杂度: O(nlogn)看上去是比较可行的,1e5不会t!盘他

接下来思考如何建边,再观察边权,两个点之间的边权居然是它们同系坐标里的最小值,那如果每个点都去暴力的求三维的边理应会有\frac{n*(n-1)}{2}条边,显然直接爆炸了,不可行,因为每次两个点相连其实只用到了一条边,所以思考如何优化?

所以我们可以以三维来排序,一共排三次,每次排完都将相邻的两个点连一条边。如果我们使用了不符合题设的边作为答案,那么它一定可以被符合题设的边替代。

那么为啥排序后相邻的两个点一定最优呢?其实不难想用贪心的思考方式,排序之后假设只看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;
}

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值