城市公交网建设问题 C++题解

城市公交网建设问题

内存限制： 256 MiB 时间限制： 1000 ms 标准输入输出 题目类型： 传统 评测方式： 文本比较

题目描述

有一张城市地图，图中的顶点为城市，无向边代表两个城市间的连通关系，边上的权为在这两个城市之间修建高速公路的造价，研究后发现，这个地图有一个特点，即任一对城市都是连通的。现在的问题是，要修建若干高速公路把所有城市联系起来，问如何设计可使得工程的总造价最少？

输入格式

n（城市数，1<≤n≤100）

e（边数）

以下e行，每行3个数i,j,wij，表示在城市i,j之间修建高速公路的造价。

输出格式

n-1行，每行为两个城市的序号a b(a<=b)，表明这两个城市间建一条高速公路。

输出顺序如下：

输出每条边时按照a < b的格式输出，优先按照a从小到大排列，若a相等，则按照b从小到大排列

样例

样例输入

复制5 8
1 2 2
2 5 9
5 4 7
4 1 10
1 3 12
4 3 6
5 3 3
2 3 8

样例输出

复制1 2
2 3
3 4
3 5

#include <bits/stdc++.h>
#define PII pair<int, int>
using namespace std;
const int N = 1e2 + 5;
int n, e, a, b, w;
bool visit[N];
vector<PII> ans;
vector<PII> G[N];
void prim() {
	memset(visit, 0, sizeof(visit));
	visit[1] = 1;
	for(int i = 2; i <= n; i++) {
		int p = 0, q = 0, mi = INT_MAX;
		for(int j = 1; j <= n; j++) {
			if(!visit[j]) {
				continue;
			} 
			for(auto k : G[j]) {
				int u = k.first, v = k.second;
				if(!visit[v] && u < mi) {
					p = j, q = v, mi = u;
				} 
			} 
		} 
		visit[q] = 1;
		ans.push_back({min(p, q), max(p, q)});
	} 
} 
int main() {
	scanf("%d %d", &n, &e);
	for(int i = 1; i <= e; i++) {
		scanf("%d %d %d", &a, &b, &w);
		G[a].push_back({w, b});
		G[b].push_back({w, a});
	} 
	prim();
	sort(ans.begin(), ans.end());
	for(auto i : ans) {
		printf("%d %d\n", i.first, i.second);
	} 
	return 0;
}