最小生成树之Kruskal---POJ1258---Agri-Net

题目链接：http://poj.org/problem?id=1258

最小生成树的模板题/(ㄒoㄒ)/~~

代码如下：

#include <iostream>
#include <cstdio>
#include <vector>
#include <algorithm>
#include <cstring>
using namespace std;
int pre[110];
int T;
int find_root(int x){return pre[x]==x?x:pre[x]=find_root(pre[x]);}
void Union(int x,int y)
{
    int rx=find_root(x);
    int ry=find_root(y);
    if(rx!=ry)
        pre[ry]=rx;
    else
        return ;
}
struct Edge
{
    int st,ed,v;
    Edge(int s,int e,int vl):st(s),ed(e),v(vl){}
};
vector<Edge> E;
bool cmp(Edge x,Edge y)
{
    return x.v<y.v;
}
int kruskal()
{
    int sum=0,num=0;
    for(int i=0;i<E.size();i++)
    {
        int a=E[i].st,b=E[i].ed,value=E[i].v;
        if(find_root(a)!=find_root(b))
        {
            num++;
            sum+=value;
            Union(a,b);
        }
        if(num==(T-1))return sum;
    }
    return sum;
}
int main()
{
    while(scanf("%d",&T)!=EOF)
    {
        memset(pre,0,sizeof(pre));
        E.clear();
        for(int i=0;i<T;i++)
        {
            pre[i]=i;
        }
        for(int i=0;i<T;i++)
        {
            for(int j=0;j<T;j++)
            {
                int val;
                scanf("%d",&val);
                E.push_back(Edge(i,j,val));
            }
        }
        sort(E.begin(),E.end(),cmp);
        int ans=kruskal();
        printf("%d\n",ans);
    }
    return 0;
}

仅代表个人观点，欢迎交流探讨，勿喷~

PhotoBy:WLOP

http://weibo.com/wlop