您的位置:首页 > 其它

POJ 1861

2016-10-23 12:06 211 查看
/*

总结一下kruskal算法。

1.新建图G,G中拥有原图中相同的节点,但没有边
2.将原图中所有的边按权值从小到大排序
3.从权值最小的边开始,如果这条边连接的两个节点于图G中不在同一个连通分量中(使用了并查集),则添加这条边到图G中
同时可以记录路径record
4.重复3,直至图G中所有的节点都在同一个连通分量中

收获,思路很清晰,需要很小心的写


*/

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

using namespace std;

class kruskal
{
public:
int a,b,value;
}edge[15005],record[15005];

int n,m;

int f[10000];//并查集
int s[10000];//根的数目

int Max;    //最大边
int pos;    //记录遍历的边数目
int sum;    //最小生成树的值

bool cmp(kruskal a,kruskal b)
{
return a.value < b.value;
}

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

int Union(int x,int y)
{
int a = Find(x);
int b = Find(y);
if(a == b)
return 0;
else if(s[a] <= s[b]){      //使用了优化并查集
f[a] = b;
s[b] += s[a];
}
else {
f[b] = a;
s[a] += s[b];
}
return 1;
}

int main()
{
//  freopen("in.txt","r",stdin);
cin>>n>>m;

for(int i = 1;i <= m; i++)
cin>>edge[i].a>>edge[i].b>>edge[i].value;

for(int i = 1;i <= n; i++) {
f[i] = i;
s[i] = 1;
}
sum = 0;//初始化
pos = 1;
Max = 0;
sort(edge+1,edge+m+1,cmp);
for(int i = 1;i <= m; i++){
if(Union(edge[i].a,edge[i].b) == 1) {
if(edge[i].value > Max)
Max = edge[i].value;
record[pos].a = edge[i].a;
record[pos].b = edge[i].b;
pos++;
}
if(pos - 1 == n - 1)
break;
}

cout<<Max<<endl<<pos-1<<endl;
for(int i = 1;i <= pos-1; i++){
cout<<record[i].a<<" "<<record[i].b<<endl;
}
return 0;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签:  kruskal poj 算法