Networking POJ1287 最小生成树 Kruscal (根据Kruscal特点,不用判断重边)

http://poj.org/problem?id=1287

给定点以及边,这里需要注意的是50个点最多,那么边最多有50*49条,注意数组不要太小了。

Kruscal模板题

#include<iostream>
#include<math.h>
#include<stdio.h>
#include<string.h>
#include<math.h>
#include<algorithm>
using namespace std;
struct node
{
    int u;
    int v;
    int w;
} edge[2600];
int pre[100];
int n,m;
bool cmp(const node&a,const node&b )
{
    return a.w<b.w;//定义排序方法;
}
int find(int x)
{
    return x==pre[x]?x:pre[x]=find(pre[x]);
    //return
}
void unions(int x,int y)
{
    int fx=find(x);
    int fy=find(y);
    if(fx!=fy)
    {
        pre[fx]=fy;
    }
}//并查集方法;
void createGraph()
{
    for(int i=1; i<=n; i++)pre[i]=i;
    for(int i=1; i<=m; i++)
    {
        cin>>edge[i].u>>edge[i].v>>edge[i].w;
    }
}
int kruscal()
{
    int ans=0;
    sort(edge+1,edge+m+1,cmp);//排序;
    for(int i=1; i<=m; i++)
    {
        if(find(edge[i].u)!=find(edge[i].v))
        {//如果没有公共节点就合并;
            unions(edge[i].u,edge[i].v);
            ans+=edge[i].w;
        }
    }
    return ans;
}
int main()
{
    while(cin>>n>>m&&n)
    {
        createGraph();
        cout<<kruscal()<<endl;
    }
}

猜你喜欢

转载自blog.csdn.net/lanshan1111/article/details/84205206