题解 | #Jungle Roads#

Jungle Roads

http://www.nowcoder.com/practice/75c19b92d6b942f08989b335afbc73c3

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

using namespace std;

const int MAXN = 27 + 3;
const int MAXE = 75 + 5;

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

int parent[MAXN];
Edge edge[MAXE];

int Find(int x){
	if(parent[x] < 0){
		return x;
	}
	return Find(parent[x]);
}

bool Union(int x, int y){
	x = Find(x);
	y = Find(y);
	if(x != y){
		if(parent[x] > parent[y]){
			parent[x] = y;
		}else if(parent[x] < parent[y]){
			parent[y] = x;
		}else{
			parent[y] = x;
			parent[x]--;
		}
		return true;
	}
	return false;
}

int Kruskal(int n, int edgeNumber){
	sort(edge, edge + edgeNumber);
	memset(parent, -1, sizeof(parent));
	int sum = 0;
	for(int i=0; i<edgeNumber; i++){
		Edge current = edge[i];
		if(Union(current.from, current.to)){
			sum += current.length;
		}
	}
	return sum;
}

int main(){
	int n;
	string s;
	int x;
	while(scanf("%d", &n) != EOF){
		int k = 0;//用来生成edge[] 
		int ntmp = n-1;
		while(ntmp--){
			cin >> s;
			scanf("%d", &x);
			int left = s[0] - 'A';
			int xnum = x;
			while(xnum--){
				cin >> s;
				scanf("%d", &x);
				int right = s[0] - 'A';
				edge[k].from = left;
				edge[k].to = right;
				edge[k++].length = x;
			}	
		}
		printf("%d\n", Kruskal(n, k));
	}
	return 0;
}










全部评论

相关推荐

不愿透露姓名的神秘牛友
07-11 11:00
点赞 评论 收藏
分享
Lorn的意义:你这标个前端是想找全栈吗?而且项目确实没什么含金量,技术栈太少了,边沉淀边找吧 现在学院本想就业好一点四年至少得高三模式两年加油吧
点赞 评论 收藏
分享
昨天 12:29
门头沟学院 Java
后端岗,实习三周感觉有点想跑路了,担心秋招被拉黑,有没有佬是字节HR知道情况的
从零开始的转码生活:你实习三周都想跑路,将来拿到offer真的愿意在这干十几二十年吗
投递字节跳动等公司8个岗位
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务