最小生成树Kruskal算法
Time Limit: 1000MS | Memory Limit: 65536K | |||
Total Submissions: 112 | Accepted: 61 |
Description
某省调查城镇交通状况,得到现有城镇道路统计表,表中列出了连接两个城镇需要花费的代价。省政府“畅通工程”的目标是使全省任何两个城镇间都可以实现交通(但不一定有直接的道路相连,只要互相间接通过道路可达即可)。问最少花费多少代价就可以完成工程?
Input
输入包含多组数据,对于每组测试数据
第一行包含两个正整数N和M(0 <=N <=1000,0 < M < 5000),分别代表现有城镇的数目和已修建的道路的数目。城镇分别以1~N编号。
接下来是M行道路信息。每一行有三个整数A,B,X(0 < A,B <= N ,0 < X < 10000),表示可以在城镇A和城镇B之间建一条花费为X的双向道路。保证数据可以完成工程。
第一行包含两个正整数N和M(0 <=N <=1000,0 < M < 5000),分别代表现有城镇的数目和已修建的道路的数目。城镇分别以1~N编号。
接下来是M行道路信息。每一行有三个整数A,B,X(0 < A,B <= N ,0 < X < 10000),表示可以在城镇A和城镇B之间建一条花费为X的双向道路。保证数据可以完成工程。
Output
对于每组数据,输出完成工程需要花费的最少代价。
Sample Input
3 4 1 2 3 2 1 4 1 3 1 2 3 2
Sample Output
3
Source
参考代码:
#include<stdio.h>
#include<iostream>
#include<algorithm>
using namespace std;
#define maxm 5050
struct node
{
int x,y,w;
}e[maxm];
int fa[maxm];
int N,M;
int ans;
int find(int x)
{
return fa[x]==x?x:fa[x]=find(fa[x]);
}
bool cmp(node a,node b)
{
return a.w<b.w;
}
int kruscal()
{
sort(e+1,e+M+1,cmp);
int cnt=N;
int i;
for(i=1;i<=N;i++)
fa[i]=i;
for(i=1;i<=M;i++)
{
int t1=find(e[i].x);
int t2=find(e[i].y);
if(t1!=t2)
{
fa[t1]=t2;
ans+=e[i].w;
cnt--;
if(cnt==1)
break;
}
}
return ans;
}
int main()
{
while(~scanf("%d%d",&N,&M))
{
int i;
for(i=1;i<=M;i++)
scanf("%d%d%d",&e[i].x,&e[i].y,&e[i].w);
printf("%d\n",kruscal());
}
return 0;
}