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.
————————————————————————————————
最小支撑树也就是最小生成树，可以用prim算法、kruskal算法，我这里用了kruskal算法，自己最开始还是没写出来…（怪我图这一块没好好学…

借鉴大佬博客之后完成，最近刷题第二次用并查集，比较熟悉了，详细见代码注释。
注意，pta这个题用scanf 和printf 会超时，要用cin 和cout，有大佬知道为啥请指教。

#include <bits/stdc++.h>
using namespace std;

typedef struct{
	int u;//顶点1
	int v;//顶点2 
	int w;//权值 
}Edge;//结构体存边的信息 
int a[201];

bool cmp(Edge edge1,Edge edge2){
	return edge1.w<edge2.w;
}

void init(){//初始化 
	for(int i=0; i<201; i++){
		a[i] = i;
	}
}

int find(int x){//查找 
	int r = x;
	while(a[r]!=r)
		r = a[r];
	return r;
}

int join(int x,int y){//合并 
	int fx = find(x);
	int fy = find(y);
	if(fx!=fy){
		a[fy] = fx;
		return 1;
	}
	return 0;
}

int main(){
    int n,e;
    while(cin >> n >> e){
    	Edge edge[e];
    	init();
    	for(int i=0; i<e; i++){
    		cin >> edge[i].u >> edge[i].v >> edge[i].w;
		}
		sort(edge,edge+e,cmp); //边按权值从小到大排序 
		int cnt=0, w_sum=0; //边数、权值和
		for(int i=0; i<e; i++){
			if(join(edge[i].u,edge[i].v)){//如果顶点u和v不在一个连通图里
				cnt++;
				w_sum += edge[i].w;
			}
			if(cnt==n-1) //边数 = 顶点数-1，找到最小生成树 
				break;
		}
		if(cnt==n-1)
			cout << w_sum << endl;
		else cout << "There is no minimum spanning tree." << endl; 
	}
    
    return 0;
}