图结构练习——最小生成树
Time Limit: 1000MS Memory Limit: 65536KB
Problem Description
有n个城市,其中有些城市之间可以修建公路,修建不同的公路费用是不同的。现在我们想知道,最少花多少钱修公路可以将所有的城市连在一起,使在任意一城市出发,可以到达其他任意的城市。
Input
输入包含多组数据,格式如下。
第一行包括两个整数n m,代表城市个数和可以修建的公路个数。(n <= 100, m <=10000)
剩下m行每行3个正整数a b c,代表城市a 和城市b之间可以修建一条公路,代价为c。
Output
每组输出占一行,仅输出最小花费。
Example Input
3 2
1 2 1
1 3 1
1 0
Example Output
2
0
Hint
解题思路同3362
#include<cstdio>
#include<iostream>//必须包含此头文件,否则超时
#include<cstring>
#include<algorithm>
using namespace std;
struct link_road
{
int a,b,cost;
}p[3500];
int father[1500];
int cmp(struct link_road p1,struct link_road p2)
{
return p1.cost<p2.cost;
}
int finda(int i)
{
while(i!=father[i])
i=father[i];
return i;
}
void mergea(int a,int b)
{
int fa=finda(a);
int fb=finda(b);
if(fa!=fb)
father[fa]=fb;
}
int main()
{
int n,m;
while(scanf("%d %d",&n,&m)!=EOF)
{
memset(p,0,sizeof(p));
for(int i=1;i<=n;i++)
father[i]=i;
for(int i=0;i<m;i++)
scanf("%d %d %d",&p[i].a,&p[i].b,&p[i].cost);
sort(p,p+m,cmp);
int sum=0;
for(int i=0;i<m;i++)
{
if(finda(p[i].a)!=finda(p[i].b))
{
mergea(p[i].a,p[i].b);
sum+=p[i].cost;
}
}
printf("%d\n",sum);
}
return 0;
}