#2020.02.05训练题解#最小生成树入门（A题）

题源HDU-1863

HDU-1866-畅通工程

Description
省政府“畅通工程”的目标是使全省任何两个村庄间都可以实现公路交通（但不一定有直接的公路相连，只要能间接通过公路可达即可）。经过调查评估，得到的统计表中列出了有可能建设公路的若干条道路的成本。现请你编写程序，计算出全省畅通需要的最低成本。

Input
测试输入包含若干测试用例。每个测试用例的第1行给出评估的道路条数 N、村庄数目M ( < 100 )；随后的 N
行对应村庄间道路的成本，每行给出一对正整数，分别是两个村庄的编号，以及此两村庄间道路的成本（也是正整数）。为简单起见，村庄从1到M编号。当N为0时，全部输入结束，相应的结果不要输出。

Output
对每个测试用例，在1行里输出全省畅通需要的最低成本。若统计数据不足以保证畅通，则输出“?”。

Sample Input
3 3
1 2 1
1 3 2
2 3 4
1 3
2 3 2
0 100

Sample Output
3
?

题意

• 多组输入，每组先输入一个N和M，N代表道路条数，M代表村庄数目，N为零时程序结束
• 因为习惯N作为点的数目，所以下方代码输入的时候调换了N和M的意义，保持习惯性输入
• 随后几行是每条道路连通的两个村庄，以及道路成本
• 输出全省畅通需要的最低成本，如果无法让全省畅通，则输出?

题解

• 这是最小生成树的模板题，分别输入对应量，调用函数即可
• 在Kruskal()函数中，用cnt统计连通了几条道路
• 如果需要全省畅通形成最小生成树，cnt应当为n-1，否则得输出?

涉及知识点

• 最小生成树MST 算法（此题用到Kruskal算法）
• 对于最小生成树的算法-详见链接博客介绍最小生成树

AC代码

#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
const int inf=0x3f3f3f3f;
const int maxn=1e3+10;
int father[maxn];
int n,m,sum,cnt;
struct node
{
	int st;
	int ed;
	int w;
}edge[maxn];
bool cmp(node x,node y)
{
	return x.w<y.w;
}
void init()
{
	for(int i=1;i<=n;i++) father[i]=i;
	memset(edge,0,sizeof(edge));
}
int find(int x)
{
	return x==father[x]?x:father[x]=find(father[x]);
}
void kruskal()
{
	sum=0,cnt=0; 
	for(int i=1;i<=m;++i)
	{
		int fx=find(edge[i].st);
		int fy=find(edge[i].ed);
		if(fx==fy) continue;//判断其是否在同一个连通分量
		sum+=edge[i].w;
		father[fx]=fy;
		cnt++;//统计有多少条边，每合并一次就是多一条边，最小生成树n-1条边 
		if(cnt==n-1) return;
	}
}
int main()
{
	while(~scanf("%d %d",&m,&n)&&m)
	{
		init();
		for(int i=1;i<=m;i++) scanf("%d %d %d",&edge[i].st,&edge[i].st,&edge[i].w);
		sort(edge+1,edge+m+1,cmp);//对边按照权值进行排序
		kruskal();
		if(cnt==n-1) printf("%d\n",sum);
		else printf("?\n");
	}
}