【YbtOJ 最小生成树 - 2】新的开始

新的开始

---

题目

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

为了保证电力的供应，小 $F$ 想到了两种办法：在这一口矿井上建立一个发电站，费用为 $v$（发电站的输出功率可以供给任意多个矿井）

将这口矿井与另外的已经有电力供应的矿井之间建立电网，费用为$p$

小 $F$ 希望你帮他想出一个保证所有矿井电力供应的最小花费

---

输入

输出

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

---

输入样例

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

输出样例

9

样例解释

---

解题思路

我们用$prim$来做这道题，设$0$号节点为超级电源，然后让矿井与$0$号节点连边，就转换成让$0n$号节点联通的花费

---

程序如下

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

using namespace std;

int n, Min, ans;

int f[1001][1001], r[100001];

bool a[100001];

int main()
{
	scanf("%d",&n);
	for(int i = 1; i <= n; ++i)
	{
		scanf("%d",&f[0][i]);
		f[i][0] = f[0][i];
	} 
	for(int i = 1; i <= n; ++i)
		for(int j = 1; j <= n; ++j)
			scanf("%d",&f[i][j]);
	for(int i = 1; i <= n; ++i)
		f[i][i] = 2147483647;
	r[0] = 0;
	a[0] = true;
	for(int i = 1; i <= n; ++i)
		r[i] = f[0][i];
	for(int i = 1; i <= n; ++i)
	{
		Min = 2147483647;
		int k = 0;
		for(int j = 1; j <= n; ++j)
		{
			if((!a[j]) && (r[j] < Min))
			{
				Min = r[j];
				k = j;
			}
		}
		ans += r[k];
		a[k] = true;
		for(int j = 1; j <= n; ++j)
		{
			if(f[k][j] < r[j])
				r[j] = f[k][j];
		}
	}
	printf("%d",ans);	
	return 0;
}