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

题目大意:输入一个整数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],否则会WA

struct 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();
	}
}