(step6.1.1)hdu 1879(继续畅通工程——最小生成树、kruscal)

来源：互联网发布：中国少儿网络美术馆编辑：程序博客网时间：2024/05/19 09:02

题目大意:输入一个整数n，表示有n个村庄。在接下来的n(n-1)/2行中，每行有4个整数begin end weight flag。分别表示从begin到end之间可以连通

，他们之间的费用为weight，flag表示这条路是否已经修好。求修路让所有村庄都连起来的最小费用

解题思路:最小生成树

代码如下:

/* * 1879_4.cpp * *  Created on: 2013年8月25日 *      Author: Administrator */#include <iostream>using namespace std;const int maxn = 110;int father[110000];//注意这里不要只开到father[110],否则会WAstruct edge{int begin;int end;int weight;};edge e[maxn*maxn];//因为m的取值是n(n-1)/2,所以他应该是maxn*maxn的形式int n;bool compare(const edge& a , const edge& b){return a.weight < b.weight;}int find_set(int x){if( x == father[x]){return x;}father[x] = find_set(father[x]);return father[x];}void kruskal(){int i;int sum = 0;//初始化for(i = 1 ; i <= n ;++i){father[i] = i;}//合并for(i = 1 ; i <= n ; ++i){int fx = find_set(e[i].begin);int fy = find_set(e[i].end);if(fx != fy){father[fx] = fy;sum += e[i].weight;}}printf("%d\n",sum);}void init(){int i,t;for( i = 1 ; i <= n ; ++i){scanf("%d%d%d%d",&e[i].begin,&e[i].end,&e[i].weight,&t);if(t){e[i].weight = 0;}}}int main(){while(scanf("%d",&n)!=EOF,n){//这种写法会WA//int m = n*(n - 1)/2;////int i;//for( i = 1 ; i <= m ; ++i){//int flag ;//scanf("%d%d%d%d",&e[i].begin,&e[i].end,&e[i].weight,&flag);////if(flag){//e[i].weight = 0;//}//}n = n*( n - 1)/2;init();sort(e + 1,e + n + 1 , compare);kruskal();}}