最优布线问题 C++题解

最优布线问题

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

题目描述

学校有n台计算机，为了方便数据传输，现要将它们用数据线连接起来。两台计算机被连接是指它们有数据线连接。由于计算机所处的位置不同，因此不同的两台计算机的连接费用往往是不同的。

当然，如果将任意两台计算机都用数据线连接，费用将是相当庞大的。为了节省费用，我们采用数据的间接传输手段，即一台计算机可以间接的通过若干台计算机（作为中转）来实现与另一台计算机的连接。

现在由你负责连接这些计算机，任务是使任意两台计算机都连通（不管是直接的或间接的）。

输入格式

第一行为整数n（2 ≤ n ≤ 100），表示计算机的数目。此后的n行，每行n个整数。第x+1行y列的整数表示直接连接第x台计算机和第y台计算机的费用。

输出格式

一个整数，表示最小的连接费用。

样例

样例输入

复制3
0 1 2
1 0 1
2 1 0

样例输出

复制2

数据范围与提示

注：表示连接1和2，2和3，费用为2。

#include <bits/stdc++.h>
#define PII pair<int, int>
using namespace std;
const int N = 1e2 + 5;
int n, ans;
bool visit[N];
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 += mi;
	} 
} 
int main() {
	scanf("%d", &n);
	for(int x, i = 1; i <= n; i++) {
		for(int j = 1; j <= n; j++) {
			scanf("%d", &x);
			G[i].push_back({x, j});
			G[j].push_back({x, i});
		} 
	} 
	prim();
	printf("%d", ans);
	return 0;
}