关闭

神奇的最小生成树........

60人阅读 评论(0) 收藏 举报

这其实是是prim算法
来解释什么是prim吧!
其实有点像贪心,每次选最短的链接………..
最后总为最小的…..
还有这是POJ 2395 Out of Hay的解法

#include<iostream>
#include<algorithm>
#include<cstdio>
using namespace std;
#define MAX_E 10000+16
#define MAX_N 2000+16
int parent[MAX_N];
int height[MAX_N];
void init(const int& n)
{
    for(int i=0;i<n;i++)
    {
        parent[i]=i; //初始化并查集
        height[i]=0; //初始化深度
    }
}
int find(const int& x)
{
    if(parent[x]==x)
        return x;
    else
        return parent[x]=find(parent[x]);
}
void unite(int x, int y)
{
    x=find(x);
    y=find(y);
    if(x==y)
        return;
    if(height[x]<height[y]) //如果深度大于前者,则直接并上去
        parent[x]=y;  //并查集操作,并上去
    else    //heigh相当于一个优化
    {
        parent[y]=x;
        if (height[x]==height[y]) //一样则++
            ++height[x];
    }
}

bool same(const int& x, const int& y)
{
    return find(x)==find(y); //看看是否有公共祖先,否则就会形成环
}
struct edge
{
    int u, v, cost;
    edge(int u=0,int v=0,int cost=0):u(u),v(v),cost(cost) {}
    bool operator < (const edge & e2) const
    { //重载比大小,为了sort服务
        return cost<e2.cost;
    }
};
edge es[MAX_E];
int V, E;
int kruskal()
{
    sort(es,es+E);  
    init(V);
    int res=0;
    for (int i=0;i<E;++i)
    {
        edge e=es[i]; 
        if(!same(e.u,e.v)) //查找是否会有环
        {
            unite(e.u,e.v); //选取第i大的边链接
            res=max(res,e.cost); //计算结果
        }
    }

    return res;
}
int main()
{
    scanf("%d%d",&V,&E);
    for (int i=0;i<E;++i)
    {
        scanf("%d%d%d",&es[i].u,&es[i].v,&es[i].cost);
        --es[i].u; --es[i].v; //这个.........
    }
    printf("%d\n",kruskal());
    return 0;
}
0
0

查看评论
* 以上用户言论只代表其个人观点,不代表CSDN网站的观点或立场
    个人资料
    • 访问:814次
    • 积分:68
    • 等级:
    • 排名:千里之外
    • 原创:6篇
    • 转载:1篇
    • 译文:0篇
    • 评论:0条
    文章存档