新的开始（最小生成树）

题目描述

发展采矿业当然首先得有矿井，小 FF 花了上次探险获得的千分之一的财富请人在岛上挖了 n 口矿井，但他似乎忘记考虑的矿井供电问题……

为了保证电力的供应，小 FF 想到了两种办法：

1. 在这一口矿井上建立一个发电站，费用为 v（发电站的输出功率可以供给任意多个矿井）。
2. 将这口矿井与另外的已经有电力供应的矿井之间建立电网，费用为 p。

小 FF 希望身为计划首席工程师的你帮他想出一个保证所有矿井电力供应的最小花费。

输入格式

第一行一个整数 n，表示矿井总数。

第 2到n+1 行，每行一个整数，第 i 个数 vi 表示在第 i 口矿井上建立发电站的费用。

接下来为一个 n* n 的矩阵 p，其中 p{i,j} 表示在第 i 口矿井和第 j 口矿井之间建立电网的费用（数据保证有 p{i,j}=p{j,i}，且 p{i,i}=0）。

输出格式

输出仅一个整数，表示让所有矿井获得充足电能的最小花费。

样例

输入复制

4
5
4
4
3
0 2 2 2
2 0 3 3
2 3 0 4
2 3 4 0

输出复制

9

小 FF 可以选择在 4 号矿井建立发电站然后把所有矿井都不其建立电网，总花费是 3+2+2+2=9。

数据范围与提示

对于 30% 的数据：1<= n<=50；
对于 100% 的数据：1<=n<=300,0<= vi, p(i,j)<=10^5。

思路：就是将建立电网所需要的花费统一为连接0或者n+1，然后用prim进行计算即可

 

#include<bits/stdc++.h>

using namespace std;

#define MAX 0x3f3f3f

int n, w, e[505][505], d[505], book[505];
int sum = 0;

void prim() {
	d[0] = 0;
	for (int i = 1; i <= n; i++)
		d[i] = MAX;
	for (int i = 0; i <= n; i++) {
		int k_mn = -1;
		for (int j = 0; j <= n; j++) {
			if (!book[j] && (k_mn == -1 || d[k_mn] > d[j]))
				k_mn = j;
		}
		book[k_mn] = 1;
		for (int j = 1; j <= n; j++) {
			if (!book[j])
				d[j] = min(d[j], e[k_mn][j]);
		}
		sum += d[k_mn];
	}
	cout << sum << endl;
}

void solve() {
	cin >> n;
	for (int i = 1; i <= n; i++) {
		cin >> w;
		e[0][i] = e[i][0] = w;
	}
	for (int i = 1; i <= n; i++) {
		for (int j = 1; j <= n; j++) {
			cin >> e[i][j];
		}
	}
	prim();
}

int main() {
	ios::sync_with_stdio(false);
	cin.tie(0);
	solve();
	return 0;
}