图结构练习——最小生成树

图结构练习——最小生成树

Time Limit: 1000ms   Memory limit: 65536K  有疑问？点这里^_^

题目描述

 有n个城市，其中有些城市之间可以修建公路，修建不同的公路费用是不同的。现在我们想知道，最少花多少钱修公路可以将所有的城市连在一起，使在任意一城市出发，可以到达其他任意的城市。

 

输入

 输入包含多组数据，格式如下。

第一行包括两个整数n m，代表城市个数和可以修建的公路个数。(n <= 100， m <=10000)

剩下m行每行3个正整数a b c，代表城市a 和城市b之间可以修建一条公路，代价为c。

 

输出

 每组输出占一行，仅输出最小花费。

示例输入

3 21 2 11 3 11 0

示例输出

20

#include<bits/stdc++.h>using namespace std;#define INF 0x3f3f3f3fint pre[105];int m,n;int sum,flag;bool visit[10010];struct node{    int a;    int b;    int cost;}road[10010];int Find(int x){    int r=x;    while(r!=pre[r])        r=pre[r];    int j=x,temp;    while(pre[j]!=r)    {        temp=pre[j];        pre[j]=r;        j=temp;    }    return r;}void Merge(int x,int y){    int fx=Find(x),fy=Find(y);    if(fx!=fy)        pre[fx]=fy;}void kruskal(){    int k;    for(int i=0;i<m-1;i++)    {        int temp=INF;        for(int j=0;j<=n;j++)        {            if(temp>road[j].cost&&!visit[j]&&Find(road[j].a)!=Find(road[j].b))                temp=road[k=j].cost;        }        if(temp==INF)        {            flag=1;            break;        }        sum+=road[k].cost;        visit[k]=true;        Merge(road[k].a,road[k].b);    }}int main(){    int u,v,cost;    while(~scanf("%d%d",&m,&n))    {        sum=0;        flag=0;        memset(visit,false,sizeof(visit));        for(int i=1;i<=100;i++)            pre[i]=i;        for(int i=0;i<n;i++)        {            cin>>road[i].a>>road[i].b>>road[i].cost;        }        kruskal();        if(!flag)printf("%d\n",sum);        else printf("-1\n");    }}