【I - Agri-Net】

最新推荐文章于 2019-12-21 09:23:40 发布

Masker_43

最新推荐文章于 2019-12-21 09:23:40 发布

阅读量240

点赞数

CC 4.0 BY-SA版权

分类专栏： ACM Virtual Judge Kuangbin专题最小生成树

本文链接：https://blog.youkuaiyun.com/flash403/article/details/94996344

ACM 同时被 3 个专栏收录

120 篇文章

订阅专栏

Virtual Judge

80 篇文章

订阅专栏

Kuangbin专题

80 篇文章

订阅专栏

本文介绍了一个基于邻接矩阵实现的PRIM算法，用于求解最小生成树问题。通过详细的代码示例，展示了如何使用优先队列进行顶点选择，以及如何更新距离和访问标记，最终计算出最小生成树的总权重。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

思路：

裸的最小生成树邻接矩阵PRIM，没什么好说的。

代码：

0ms 688kB

//0ms		688kB


#include <iostream>
#include <cstdio>
#include <queue>
#include <cstring>
#define INF 0x3f3f3f3f

using namespace std;

const int maxn = 105;

int N;
int ans;
int vis[maxn];
int dis[maxn];
int mp[maxn][maxn];
struct NODE{
	int id;
	int dis;
	friend bool operator > (NODE a , NODE b){
		return a.dis > b.dis;
	} 
	NODE(int id,int dis) : id(id) , dis(dis) {} ;
};
priority_queue <NODE , vector<NODE> , greater<NODE> > Q;

void INIT(){
	ans = 0;
	memset(dis , INF , sizeof(dis)); 
	memset(vis , 0 , sizeof(vis));
	return ;
}

void PRIM(){
	dis[1] = 0;
	Q.push(NODE(1,0));
	while(Q.size()){
		NODE cur = Q.top() ; Q.pop() ;
		int id = cur.id;
		if(vis[id])
			continue;
		vis[id] = true;
		ans += cur.dis;
		for(int i=1;i<=N;i++)
			if(!vis[i] && dis[i] > mp[id][i]){
				dis[i] = mp[id][i];
				Q.push(NODE(i , dis[i]));
			}
	}
	return ;
}

int main(){
	while(~scanf("%d" , &N)){
		INIT();
		for(int i=1;i<=N;i++)
			for(int j=1;j<=N;j++)
				scanf("%d" , &mp[i][j]);
		PRIM();
		printf("%d\n" , ans);
	}
	return 0;
}