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

![在这里插入图片描述](https://www.icode9.com/i/ll/?i=20210327081220543.png)

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


解题思路

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

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


代码

#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);
}
上一篇:图论的一些算法总结


下一篇:随机prim算法简易全解(python实现)