Kurskal算最小生成树

#include <iostream>
#include <cstdlib>
/*含有V个顶点的最小生成树：找到v-1条边，构成
一个最小生成树，按照边的权值一次找最小的边出来，
如果没有构成环，则放到树中，如果构成环，则继续找下一条边*/
const int maxnum =100;
const int maxint = 99999;
int visited[maxnum][maxnum];//用来表示边是否被访问过
int Set[maxnum];


using namespace std;


typedef struct{
    int id;
    int dist;
}node;
typedef struct{
    node v[maxnum];
    int e[maxnum][maxnum];
    int vNum;
    int eNum;
}graph;


void creatGrapg(graph * g){
    cout<<"请输入顶点数和边数:";
    cin>>g->vNum>>g->eNum;
    for(int i =0;i<maxnum;i++)
        for(int j =0;j<maxnum;j++){
            g->e[i][j] = maxint;
    }
    int b,e,len;
    cout <<"输入边的权值："<<endl;
    for(int j=0;j<g->eNum;j++){
        cin>>b>>e>>len;
        g->e[b][e] = g->e[e][b] = len;
    }
}


void kruskal(graph * g){
    cout <<"最小生成树是:\n";
    int k =0;
    int a =0,b=0;
    int min = g->e[a][b];
    for(int i=0;i<g->vNum;i++)
        for(int j =0;j<g->vNum;j++){
            visited[i][j] = 0;
    }


    for(int i =0;i<g->vNum;i++){
        Set[i] =i;
    }
    while(k<g->vNum-1){//一共要选择v-1条边
        //找出最小的边
        for(int i=0;i<g->vNum;i++){
            for(int j=0;j<g->vNum;j++){
                if(g->e[i][j] <min&&visited[i][j]==0){
                    min = g->e[i][j];
                    a=i;
                    b=j;
                }
            }
        }
        visited[a][b] = visited[b][a] = 1;
        min = maxint;
        if(Set[a] != Set[b]){//a,b不在一个联通分量中,合并
            k++;//找到了一条合适的边
            cout <<a<<"->"<<b<<endl;
            //合并集合，将联通部件b合并到联通部件a当中
            int tmp_set = Set[b];
            for(int i=0;i<g->vNum;i++){
                if(Set[i] == tmp_set){
                    Set[i] = Set[a];//将b合并到a中
                }
                cout<<Set[i]<<" ";
            }
            cout<<endl;
        }
    }
}






int main()
{
    graph g;
    creatGrapg(&g);
    kruskal(&g);
    int i;
    for(i=0;i<g.vNum;i++)
        cout<<Set[i]<<" ";
    cout<<endl;
   system("pause");
    return 0;
}