Hdu 1863（畅通工程/继续畅通工程）（Kruskal算法）

题目链接:(http://acm.hdu.edu.cn/showproblem.php?pid=1863)

使用最小生成树算法中的Kruskal算法
代码如下

#include <iostream>//hdu1863
#include <algorithm>

using namespace std;

struct Edgenode{
    int from;//起始点
    int to;//到达点
    int s;//权值
}edge[20000];

int father[105],ans;

void init(int m)//初始化函数
{
    for(int i = 1; i <= m; i++){
        father[i] = i;
    }
}

int Find(int node)//找根节点
{
    while(node != father[node]){
        node = father[node];
    }
    return node;
}

bool cmp(Edgenode a,Edgenode b)//按照两点间权值升序排列
{
    return a.s < b.s;
}

int Kruskal(int n,int m)
{
    int count = 0;
    sort(edge,edge + n,cmp);
    for(int i = 0; i < n; i++){
        int u = Find(edge[i].from);//找根节点
        int v = Find(edge[i].to);
        if(u != v){
            father[u] = v;//合并线路
            count++;//边数加1
            ans += edge[i].s;
        }
        if(count = m - 1){
            break;
        }
    }
    return count;
}
int main()
{
    int m,n;
    while(cin>>n>>m,n != 0){
        init(m);
        ans = 0;
        for(int i = 0; i < n; i++){
            cin>>edge[i].from>>edge[i].to>>edge[i].s;
        }
        if( Kruskal(n,m) == m - 1 ){
            cout<<ans<<endl;
        }
        else{
            cout<<"?"<<endl;
        }
    }
    return 0;
}

题目链接继续畅通工程
与畅通工程不同的是，继续畅通工程则是给了一些已经存在的边，那么这些边的权值就可以记为0，那么问题就转化成了畅通工程问题，只不过其中的有些边所花费的代价为0。

#include <iostream>
#include <algorithm>
#include <cstdio>
using namespace std;

struct Edgenode{
    int from;//起始点
    int to;//到达点
    int s;//权值
}edge[20000];

int father[105],ans;

void init(int m)//初始化函数
{
    for(int i = 1; i <= m; i++){
        father[i] = i;
    }
}

int Find(int node)//找根节点
{
    while(node != father[node]){
        node = father[node];
    }
    return node;
}

bool cmp(Edgenode a,Edgenode b)//按照两点间权值升序排列
{
    return a.s < b.s;
}

void Kruskal(int m,int n)
{
    int count = 0;
    sort(edge,edge + m,cmp);
    for(int i = 0; i < m; i++){
        int u = Find(edge[i].from);//找根节点
        int v = Find(edge[i].to);
        if(u != v){
            father[u] = v;//合并线路
            count++;//边数加1
            ans += edge[i].s;
        }
        if(count == n - 1){
            break;
        }
    }
}

int main()
{
    int m,n,flag;
    while(cin>>n,n != 0){
        m = n * (n - 1) / 2;
        init(n);
        ans = 0;
        for(int i = 0; i < m; i++){
            scanf("%d%d%d%d",&edge[i].from,&edge[i].to,&edge[i].s,&flag);
            if(flag){//如果路径已存在，那么看作消耗代价为0的未建边
                edge[i].s = 0;
            }
        }
        Kruskal(m,n);
        printf("%d\n",ans);
    }
    return 0;
}