牛客网 浙江大学复试上机 继续畅通工程 最小生成树(使用并查集)的应用 计算机考研机试指南整理

继续畅通工程

分析：与上题十分相似，唯一不同的是，本题中有些道路是已经建设好的，即这些道路无需支付费用，而且已经将道路两端的城市连接了起来。因此，可将已经建好的道路的权值设为0.

#include <iostream>
#include <cstdio>
#include <algorithm>

using namespace std;

const int MAXN = 100;

struct Edge{
	int from;
	int to;
	int length;
	bool operator <(const Edge& e) const{
		return length <e.length;
	}
};

Edge edge[MAXN* MAXN];
int father[MAXN];
int height[MAXN];

void Initial(int n){
	for(int i = 0 ; i <=n; i++){ //注意 
		father[i] = i;
		height[i] = 0;
	}
}

int Find(int x){
	if(x != father[x]){
		father[x] = Find(father[x]);
	}
	return father[x];
}

void Union(int x,int y){
	x = Find(x);
	y = Find(y);
	if(x != y){
		if(height[x] < height[y]){
			father[x] = y;
		}else if(height[x] > height[y]){
			father[y] = x;
		}else {
			father[y] = x;
			height[x] ++;
		}
	}
	return ;	
}

int Kruskal(int n,int edgeNumber){
	Initial(n);
	sort(edge,edge + edgeNumber);
	int sum = 0;
	for(int i = 0 ; i < edgeNumber;i++){
		Edge current = edge[i];
		if(Find(current.from) != Find(current.to)){
			Union(current.from ,current.to);
			sum += current.length;
		}
	}
	return sum;
}


int main(){
	int n;
	while(cin>>n){
		if(n == 0){
			break;
		}
		int edgeNumber = n*(n-1) /2;
		for(int i = 0 ; i <edgeNumber;i++){
			int status;
			cin>>edge[i].from>>edge[i].to>>edge[i].length>>status;
			if(status == 1){
				edge[i].length = 0;
			}
		}
		int answer = Kruskal(n,edgeNumber);
		cout<<answer<<endl;
	}
	return 0;
}