http://blog.csdn.net/hackbuteer1/article/details/7265681
http://ac.jobdu.com/problem.php?id=1347 孤岛连通工程
刚开始的时候使用qsort排序函数进行排序提交一直都是TLE,后来无意中改为sort排序函数提交就AC了,真是太神奇了。。。
#include<iostream>
#include<algorithm>
using namespace std;
#include<stdio.h>
struct Edge
{
int x;
int y;
int cost;
}edge[10001];
int set[10001];
inline int find(int x) //带路径优化的并查集查找算法
{
int i,j,r;
r=x;
while(set[r]!=r)
r=set[r];
i=x;
while(i!=r)
{
j=set[i];
set[i]=r;
i=j;
}
return r;
}
inline void merge(int x,int y) //优化的并查集归并算法
{
int t=find(x);
int h=find(y);
if(t<h)
set[h]=t;
else
set[t]=h;
}
/*
int cmp(const void *a,const void *b)
{
return((*(struct Edge *)a).cost-(*(Edge *)b).cost);
}
*/
bool cmp(const Edge& a,const Edge& b)
{
return a.cost<b.cost;
}
void init(int n) //初始化并查集,各点为孤立点,分支数为n
{
for(int i=1;i<=n;i++)
set[i]=i;
}
int kruskal(int n,int m)
{
int i,num,sum,from,to;
//qsort(edge,m,sizeof(edge[0]),cmp);
sort(edge,edge+m,cmp);
sum=num=0;
for(i=0;i<m;i++)
{
from = find(edge[i].x); //判断线段的起始点所在的集合
to = find(edge[i].y); //判断线段的终点所在的集合
if(from != to) //如果线段的两个端点所在的集合不一样
{
merge(from,to); //合并两个集合
sum+=edge[i].cost;
num++;
}
if(num==n-1)
break;
}
if(num<n-1)
return -1;
else
return sum;
}
int main(void)
{
int i,n,m,k;
while(scanf("%d %d",&n,&m)!=EOF)
{
init(n); //初始化
for(i=0;i<m;i++)
{
scanf("%d %d %d",&edge[i].x,&edge[i].y,&edge[i].cost);
}
k=kruskal(n,m);
if(k==-1)
printf("no\n");
else
printf("%d\n",k);
}
return 0;
}
inline int find(int x) //带路径优化的并查集查找算法
{
int i,j,r;
r=x;
while(set[r]!=r)
r=set[r];//找到其所属集合的编号r
i=x;
while(i!=r)
{
j=set[i];//j为当前节点的祖先
set[i]=r;//set[i]=r;//把当前节点的编号设置为其祖先的编号
i=j;//向上遍历
}
return r;
}
inline void merge(int x,int y) //优化的并查集归并算法
{
if(x<y)
set[y]=x;
else
set[x]=y;
}
int cmp(const void* a,const void* b)
{
return ((Edge*)a)->cost>((Edge*)b)->cost?1:-1;
}
void init(int n) //初始化并查集,各点为孤立点,分支数为n
{
for(int i=1;i<=n;i++)
set[i]=i;
}
int kruskal(int n,int m)
{
int i,num,sum,from,to;
qsort(edge,m,sizeof(edge[0]),cmp);
sum=num=0;
for(i=0;i<m;i++)
{
from = find(edge[i].x); //判断线段的起始点所在的集合
to = find(edge[i].y); //判断线段的终点所在的集合
if(from != to) //如果线段的两个端点所在的集合不一样
{
merge(from,to); //合并两个集合
sum+=edge[i].cost;
num++;
}
if(num==n-1)
break;
}
if(num<n-1)
return -1;
else
return sum;
}