【Ybtoj 第11章例2】新的开始【最小生成树】

![在这里插入图片描述](https://img-blog.csdnimg.cn/20210327081220543.png)

---

解题思路

一直不知道怎么处理建立发电站的部分，然后还是看了下解题思路，然后感叹自己真的菜。。

真的是一道$prim$模板题，设$0$号节点为超级电源，在矿井上建立发电站相当于让它与$0$号节点连边。题目就被转换为了让$0n$号节点联通的花费。

---

代码

#include<iostream>
#include<cstdio>
#include<iomanip>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<bitset>
using namespace std;

int n,ans,w[310],a[310][310],dis[310],v[310];

int main(){
	scanf("%d",&n);
	for(int i=1;i<=n;i++)	
		scanf("%d",&a[1][i+1]);
	for(int i=1;i<=n;i++)
		for(int j=1;j<=n;j++)
			scanf("%d",&a[i+1][j+1]);
	memset(dis,0X7f,sizeof(dis));
	dis[1]=0; 
	for(int i=1;i<=n+1;i++)
	{
		int k=0;
		for(int j=1;j<=n+1;j++)
		{
			if(!v[j]&&dis[j]<dis[k])
				k=j;
		}
		if(!k)
			break;
		v[k]=1;
		ans+=dis[k];
		for(int j=1;j<=n+1;j++)
		{
			if(a[k][j]<dis[j])
				dis[j]=a[k][j];
		}
	}
	printf("%d",ans);
}