HDU——1879 继续畅通工程（最小生成树问题）

原题链接：http://acm.hdu.edu.cn/showproblem.php?pid=1879

样例：

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

Sample Output
3
1
0

题意： 给定n个村庄数目，以及村庄之间的信息，问连通这些村庄的最小成本。

解题思路： 这道题给出的信息为边，所以我们用Kruskal算法来写。对于输入数据要注意的是有一个标志flag表示其中两个村庄是否修建。如果已经修建我们要将之间的消耗成本更改为0。其他的和Kruskal算法模板一样，若对Kruskal算法不填熟悉的话，这里指路一篇blog：https://blog.csdn.net/hzf0701/article/details/107933639。

AC代码：

/*
*邮箱：unique_powerhouse@qq.com
*blog:https://me.csdn.net/hzf0701
*注：文章若有任何问题请私信我或评论区留言，谢谢支持。
*
*/
#include<bits/stdc++.h>	//POJ不支持

#define rep(i,a,n) for (int i=a;i<=n;i++)//i为循环变量，a为初始值，n为界限值，递增
#define per(i,a,n) for (int i=a;i>=n;i--)//i为循环变量， a为初始值，n为界限值，递减。
#define pb push_back
#define IOS ios::sync_with_stdio(false);cin.tie(0); cout.tie(0)
#define fi first
#define se second
#define mp make_pair

using namespace std;

const int inf = 0x3f3f3f3f;//无穷大
const int maxn = 1e5;//最大值。
typedef long long ll;
typedef long double ld;
typedef pair<ll, ll>  pll;
typedef pair<int, int> pii;
//*******************************分割线，以上为自定义代码模板***************************************//

struct Edge{
	int u,v;//两个村庄的编号。
	int w;//两个村庄之间的成本。
	bool operator <(const Edge &a){
		return w<a.w;//运算符重载。
	}
};
Edge edges[maxn];
int verx[maxn];//verx[i]代表i村庄所属的连通分量。
int n,cnt;//n个村庄，cnt代表边数。
int u,v,w;//临时变量。
bool flag;
void Kruskal(){
	int sum=0;
	rep(i,1,n)verx[i]=i;//初始连通分量为它自己。
	sort(edges,edges+cnt);
	rep(i,0,cnt-1){
		u=verx[edges[i].u];
		v=verx[edges[i].v];
		if(u!=v){
			rep(j,1,n){
				if(verx[j]==v)verx[j]=u;
			}
			sum+=edges[i].w;
		}
	}
	cout<<sum<<endl;
}
int main(){
	//freopen("in.txt", "r", stdin);//提交的时候要注释掉
	IOS;
	while(cin>>n&&n){
		cnt=n*(n-1)/2;
		rep(i,0,cnt-1){
			cin>>u>>v>>w>>flag;
			edges[i].u=u;edges[i].v=v;edges[i].w=w;
			if(flag){
				edges[i].w=0;
			}
		}
		Kruskal();
	}
	return 0;
}