Kruskal算法模块

Kruskal算法

prim算法链接

#include <iostream>
#include <cstring>
#include <algorithm> 
using namespace std;
#define MAX 100
#define INF 0x3f3f3f3f
struct Road
{
    int a, b; //a, b为一条边的两个顶点 
    int w;  //边的权值 
}road[MAX];

struct Vex{
    int x;
    int y;
}vex[MAX];

bool cmp(Road a, Road b){
    return a.w < b.w;
}
int N, E, k = 0;
int v[MAX];  //定义并查集数组
int getRoot(int a)
{
    while(a != v[a])
        a = v[a];
    return a;
} 

void unionn(int r1, int r2){
    v[r2] = r1;
}

void Kruskal(int &sum, Road road[], Vex vex[])
{
    int i, a, b;
    sum = 0;
    for(i = 0; i < N; i++)
    {
        v[i] = i;
    } 
    sort(road, road + E, cmp); //对road数组的E条边按权值从小到大排序 
    for(i = 0; i < E; ++i)
    {
        a = getRoot(road[i].a);
        b = getRoot(road[i].b);
        if(a != b)
        {
            unionn(a, b);
            vex[k].x = road[i].a;
            vex[k].y = road[i].b;
            k++;
            sum += road[i].w; //求生成树的权值。若是将生成树的各边输出，可以存放在数组中 
        } 
    }
}

int main() {
    int x, y, w;
    cin >> E >> N;
    for(int i = 0; i < E; i++)
    {
        cin >> x >> y >> w;
        road[i].a = x;
        road[i].b = y;
        road[i].w = w;
    }
    int sum = 0;
    Kruskal(sum, road, vex);
    for(int i = 0; i < k; i++)
    {
        cout << vex[i].x << " " << vex[i].y << endl;
    }
    cout << sum << endl;
    return 0;
}
/*算法思想：（适用于稀疏图）
将图中边按照权值从小到大排序，然后从最小边开始扫描各边，并检测当前边是否为候选边
，即是否该边的并入会构成回路，如不构成回路，则将该边并入当前生成树中，直到所有边都
检测完毕为止。 
*/

/*测试数据
8 5
0 1 5
0 2 1
0 3 2
1 2 3
1 4 4
2 3 6
2 4 2
3 4 3
*/