数据结构:kruskal算法

#include <stdio.h>
#include <stdlib.h>
#include <algorithm>
using namespace std;

typedef struct {
    int begin;
    int end;
    int weight;
} Edge;
int father[1005];

void init() {
    for (int i=0; i<1005; i++) {
        father[i] = i;
    }
}

int find(int x) {
    return father[x] == x ? x : father[x] = find(father[x]);
}

void join (int x, int y) {
    int p1 = find(x);
    int p2 = find(y);

    if (p1 == p2) {
        return ;
    }
    father[p1] = p2;
}

void kruskal (Edge edges[], int n, int m) {
    int sum_weight = 0;
    int num = 0;
    init();
    for (int i=0; i<m; i++) {
        if (find(edges[i].begin) != find(edges[i].end)) {
            printf("%d %d %d\n", edges[i].begin, edges[i].end, edges[i].weight);
            sum_weight += edges[i].weight;
            num++;
            join(edges[i].begin, edges[i].end);
        }
        if (num >= n-1) { break; }
    }
    printf("weight of MST is %d\n", sum_weight);

}

bool cmp (Edge a, Edge b) {
    return a.weight < b.weight;
}

int main (int argc, char **argv) {
    int n, m;//顶点、边
    Edge edges[20];
    scanf("%d %d", &n, &m);
    for (int i=0; i<m; i++) {
        scanf("%d %d %d", &edges[i].begin, &edges[i].end, &edges[i].weight);
    }
    sort(edges, edges+m, cmp);
    printf("kruskal:\n");
    kruskal(edges, n,  m);
    return 0;
}
/*
9 15
4 7 7
2 8 8
0 1 10
0 5 11
1 8 12
3 7 16
1 6 16
5 6 17
1 2 18
6 7 19
3 4 20
3 8 21
2 3 22
3 6 24
4 5 26
*/
/*
99
*/

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值