7-5 最小支撑树 (10 分)

7-5 最小支撑树 (10 分)

给定一个包含n个顶点的正权无向图，编号为0至n-1。请编写程序求出其最小支撑树，并计算其代价。

输入格式:

输入包含多组数据。每组数据第一行为2个整数n和e，均不超过200，分别表示图的顶点数和边数。接下来e行表示每条边的信息，每行为3个非负整数a、b、c，其中a和b表示该边的端点编号，c表示权值。各边并非按端点编号顺序排列。

输出格式:

对于每组数据，若存在最小支撑树则输出一个整数，为最小支撑树各边权值之和；若不存在最小支撑树，则输出“There is no minimum spanning tree.”。

输入样例:

4 5
0 1 1
0 3 1
1 3 5
1 2 1
2 3 8
4 2
0 1 1
2 3 8

输出样例:

3
There is no minimum spanning tree.

代码+思路

记录每一条边的起点、终点、权重，对边从小到大排序然后依次遍历。若该边起点终点不是同一起点，则可以加入，边数+1，否则会形成环路。若边数为点数-1，则成功，否则就没有最小支撑树。

#include<bits/stdc++.h>
using namespace std;
int F[201];
int G[201][201];
int find(int x){
    if(F[x]==x) return F[x];
    else return F[x]=find(F[x]);
}
typedef struct{
    int s,e,w;
}Edge;
bool cmp(Edge e1, Edge e2){ 
    return e1.w < e2.w; 
}
bool Func(int a,int b){
    int x=find(a),y=find(b);
    if(x!=y){
        F[y]=x;return true;
    } 
    return false;
}
int main(){
    Edge edge[201];
    int n,e,cnt=0,sum=0;
    while(cin>>n>>e){
        for(int i=0;i<e;i++){
            cin>>edge[i].s>>edge[i].e>>edge[i].w;
        }
        for(int i=0;i<n;i++) F[i]=i;
        sort(edge,edge+e,cmp);
        for(int i=0;i<e;i++){
            if(Func(edge[i].s,edge[i].e)){
                cnt++;
                sum+=edge[i].w;
            }
            if(cnt==n-1) break;
        }
        if(cnt==n-1) printf("%d\n",sum);
        else cout<<"There is no minimum spanning tree."<<endl;
    }
}