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

	}

}