kruscal最小生成树算法

// kruscal.cpp : 定义控制台应用程序的入口点。
//

#include "stdafx.h"
#include <queue>
#include <functional>
#include <iostream>
#include <vector>
using namespace std;
class  UFset
{
public:
vector<int> parent;
UFset(int n);
int  Find( int );
void  Union( int ,  int );
} ;

UFset::UFset(int n)
{
parent.resize(n, -1);//根节点放置的是当前树上的节点数量的负值，初始化是为-1，表示只有一个元素
}

int  UFset::Find( int  x)
{
if  (parent[x]  <=   0 )
return  x;
else
{
parent[x]  =  Find(parent[x]);
return  parent[x];
} //  压缩路径
}

void  UFset::Union( int  x,  int  y)
{
int  pX  =  Find(x);   // 根节点的位置
int  pY  =  Find(y);
int  tmp;

tmp  =  parent[pX]  +  parent[pY];  //  加权合并

if  (pX  !=  pY)//不在一个集合中
{
if  (parent[pX]  >  parent[pY])
{
parent[pX]  =  pY;
parent[pY]  =  tmp;
}
else
{
parent[pY]  =  pX;
parent[pX]  =  tmp;
}
}

}

struct Edge
{
int dis;
int v,w;
bool operator<(const Edge &e1)const
{
return dis<e1.dis;
}
bool operator>(const Edge &e1)const
{
return dis>e1.dis;
}
Edge(int vv,int ww,int diss)
{
v=vv;
w=ww;
dis=diss;
}
};

int _tmain(int argc, _TCHAR* argv[])
{

priority_queue<Edge,vector<Edge>,greater<Edge>> e;//边集
priority_queue<Edge,vector<Edge>,less<Edge>> te;//生成树的边集
int p, t;//顶点数,边数

FILE *fp;
fp=fopen("data.txt", "r");
fscanf(fp,"%d %d", &p, &t);

int i;
for(i =0; i < t; i++)
{
int t1, t2, t3;

fscanf(fp, "%d %d %d", &t1, &t2, &t3);
Edge x(t1, t2, t3);
e.push(x);
}

UFset u(p+1);

while(e.size())
{
Edge x=e.top();
e.pop();
int a=u.Find(x.w);
int b=u.Find(x.v);
if (a!=b)
{
te.push(x);
cout<<x.v <<"-"<<x.w<<":"<<x.dis<<endl;
u.Union(a,b);
}
}

return 0;
}

6 10
1 2 6
1 3 1
1 4 5
2 3 5
2 5 3
3 4 5
3 5 6
3 6 4
4 6 2
5 6 6