POJ 1251 Jungle Roads 九度 OJ1154 最小生成樹問題 輸入轉換

題目鏈接

題目大意:
給你無向圖中每條邊的長度,要你求該圖的最小生成樹。其中每個頂點用大寫字母A-Z表示。

解題思路:
最小生成樹問題,這裏使用的是Kruskal算法,注意轉換輸入格式。

筆記:scanf()函數輸入的時候,%s比%c使用起來更加方便,因爲%s不需要考慮空格和換行符的問題

AC代碼:將scanf函數的輸入全部換成註釋掉的cin輸入也可以AC

#include<iostream>
#include<math.h>
#include<algorithm>
#include<stdio.h>
using namespace std;
#define maxn 27
int tree[maxn];
int findroot(int x) {
	if (tree[x] == -1)return x;
	int tmp = findroot(tree[x]);
	tree[x] = tmp;
	return tmp;
}
bool unit(int a, int b) {
	int fa = findroot(a);
	int fb = findroot(b);
	if (fa != fb) {
		tree[fa] = fb;
		return true;
	}
	return false;
}
struct E {
	int from, to;
	int cost;
	bool operator < (const E &a) const {
		return cost < a.cost;
	}
}edge[76];

void init() {
	for (int i = 0; i < maxn; i++) {
		tree[i] = -1;
	}
}
int main() {
	int n;
	//while (cin >> n && n) {
	while (scanf("%d",&n)!=EOF && n) {
		init();
		int cnt = 0;
		while (n - 1 > 0) {
			char a[2], b[2]; int cost, m;
			//cin >> a >> m;
			scanf("%s %d ", a, &m);
			while (m--) {
				//cin >> b >> cost;
				scanf("%s %d", b, &cost);
				edge[++cnt].from = a[0] - 'A';
				edge[cnt].to = b[0] - 'A';
				edge[cnt].cost = cost;
			}
			n--;
		}
		sort(edge + 1, edge + 1 + cnt);
		int ans = 0;
		for (int i = 1; i <= cnt; i++) {
			if (unit(edge[i].from, edge[i].to)) {
				ans += edge[i].cost;
			}
		}
		printf("%d\n", ans);
	}
	return 0;
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章