镖局运镖(最小生成树 Kruskal)

7-11 镖局运镖 (30 分)

镖局的运镖,就是运货(类似现在的物流)。镖局每到一个新地方开展业务,都需要对运镖途中的绿林好汉进行打点。好说话的打点费就比较低,不好说话的打点费就比较高。龙门镖局现在有一趟镖请你来规划路线,已知城市的地图,你需要选择一些道路进行疏通,以便镖局可以到达任意一个城市,要求花费的银子越少越好。

输入格式:

第一行有两个数n和m,n表示有n个城市(编号从1到n),m表示有m条道路。接下来m行,每行形如“a b c”用来表示一条道路,意思是城市a到城市b连通且打点需要花费的银子数是c。

输出格式:

若通过打点能抵达所有城市,则输出最少需要花费的银子总数。若不能抵达所有的城市则输出“impossible”。

输入样例:

3 3
1 2 1
1 3 2
2 3 4

输出样例:

3

题目大意:一个无向图有n个点,给你m条线段信息,表示两点之间的距离。现在求最小的路径总长度,将这n个点连成一个连通图,其实就是求最小生成树。

最小生成树有两种算法:Kruskal和Prim算法。两种方法求出来的结果一定都是最小生成树(最小生成树可能不唯一,但是最短总路径一定一样的),但是两种方法思路不一样,使用场合也不同。 

Kruskal是不断地在找最小边,Prim是找距离生成树权值最小的点。Kruskal主要找边,Prim主要找点,这就决定了当边少的时候,就用Kruskal;当点少的时候,就用Prim。专业的说,Kruskal适合稀疏图,Prim适合稠密图。

(如果想达到最佳时间,其实应该两种方法都用,先判断一下给出的是稀疏图还是稠密图。忘了在哪里看到的)

 

Kruskal思想很简单,就是将边从小到大排序,从小的开始找,如果这条边的两个点不在一棵树上,那么将这两个点连接起来;如果两个点在一棵树上,跳过,继续找下一个边......重复进行,直到增加的边的数量达到n-1条。因为n个点的最小生成树的线段一定有n-1个边。

在判断是否是一棵树上的时候,用到了并查集。写这道题的时候,将Union函数中的f[a]=b写成了a=f[b],debug的时候还在奇怪我的并查集怎么没有用??还是对并查集理解不够熟练吧。

并查集的核心

int Find(int x)
{
    if(x==f[x]) return x;
    else return f[x]=Find(f[x]); //路径压缩
}

void Union(int x,int y)
{
    int a=Find(x),b=Find(y);
    if(a!=b) f[a]=b;  
}


 for(int i=1;i<=n;i++) //初始化f数组
        f[i]=i;

这道题可能搜不到,但是网上有很多题目描述不一样,但是实质是一样的题目。这题是天梯赛学校选拔时候出的,我当时学会了Dijstra算法,想着和Prim差不多,应该可以现场写出来(我对Prim和Kruskal仅限于理解含义,代码从没写过)。浪费了30分钟在这道题上,这次让我下定决心把图论学到扎实,做到遇到图论可以有思路,不浪费时间地解决问题。

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

using namespace std;
const int Max = 1000010;

int f[Max];
int n,m,x,y;

int Find(int x)
{
    if(x==f[x]) return x;
    else return f[x]=Find(f[x]);
}

void Union(int x,int y)
{
    int a=Find(x),b=Find(y);
    if(a!=b) f[a]=b;
}

struct node
{
    int x,y,step;
};
node a[Max];

bool cmp(node a,node b)
{
    return a.step<b.step;
}

void kruskal()
{
    for(int i=1;i<=n;i++)//并查集f数组初始化
        f[i]=i;
    
    for(int i=0;i<m;i++){
        cin>>a[i].x>>a[i].y>>a[i].step;
    }
    sort(a,a+m,cmp);//按路径长度从小到大排序
    int sum=0,t=0;//sum为最小路径总和,t为已经添加了的路径条数
    for(int i=0;i<m;i++)
    {
        if(t==n-1) break;//最小生成树的路径有n-1条
        if(Find(a[i].x)!=Find(a[i].y)) {
            sum +=a[i].step;
            Union(a[i].x,a[i].y);
            t++;
        }
    }
    if(t==n-1)cout<<sum<<endl;
    else cout<<"impossible";
}

int main()
{
    cin>>n>>m;
    kruskal();
    return 0;
}

网上找的其他比较复杂的测试用例

6 10
1 2 16
1 6 21
1 5 19
2 3 5
2 4 6
2 6 11
3 4 6
6 4 14
5 4 18
5 6 33

答案:56

 

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值