WEEK6 周记 作业——最小生成树_数据中心
一、题意
1.简述
4.样例
Input
4
5
1
1 2 3
1 3 4
1 4 5
2 3 8
3 4 2
Output
4
样例解释
二、算法
主要思路
最终题意可以转化为:给定无向图求解一颗生成树,使得最大边权最小!
算法一:根据kruskal最小生成树算法,其过程就是不断添加最小的且不构成环路的边。利用这个算法求出来的生成树的最大边权一定是所有生成树中最小的,否则他就不是最小生成树了。
算法二:二分答案。先定下一个最大边权,然后看看这个边权能不能形成一个生成树。用二分法获知第一个能够形成生成树的边权,这个边权就是最小的最大边权。
检查边权能不能形成一个生成树,可以通过dfs或者bfs遍历搜索,只走边权小等于于设定的最大边权的边,看看最终生成的树是不是将所有的节点都经过了,这样的时间复杂度是
O
(
n
)
O(n)
O(n)。所以总的时间复杂度跟算法一相同,都是
O
(
n
l
o
g
n
)
O(nlogn)
O(nlogn)。
实际上,这种让最大的最小或者让最小的最大等等这类的问题,往往都存在着线性的变化,所以往往都能用二分法解决。
kruskal算法的实现除了可以通过边队列以外,还可以用一个边数组存储所有的边,然后sort一下。后一种的时间效率稍微好一点,时间常数小一点。
三、代码
#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstring>
#include<queue>
using namespace std;
int n,m,root,mmax;
struct triple{
int a,b,lth;
triple(int a,int b,int lth)
:a(a),b(b),lth(lth){}
bool operator<(const triple& t)const{
return lth>t.lth;
}
};
priority_queue<triple> q;
pair<int,int> f[50010];//<pre_node,node_num>
int find(int a){
if(f[a].first==a){
return a;
}
int root = find(f[a].first);
f[a].first = root;
return root;
}
int unit(int a,int b){
if(f[a].second<f[b].second)
swap(a,b);//保证a这棵树是节点最多的
f[b].first = a;
f[a].second += f[b].second;
return a;
}
void kruskal(){
while(!q.empty()){
triple min = q.top();
int root1 = find(min.a);
int root2 = find(min.b);
if(root1!=root2){
mmax = min.lth;
unit(root1,root2);
}
q.pop();
}
}
int main()
{
scanf("%d%d%d",&n,&m,&root);
for(int i=0;i<n;i++)
f[i] = {i,0};
for(int i=0;i<m;i++){
int a,b,tt;
scanf("%d%d%d",&a,&b,&tt);
triple t(a,b,tt);
q.push(t);
}
kruskal();
printf("%d",mmax);
return 0;
}