最小生成树Prime算法

洛谷p1546链接
Prime算法的核心也是贪心,但是不同的就是,它是一直维护一颗树,
直到变成一颗最小生成树,
在这里插入图片描述

#include<bits/stdc++.h>
using namespace std;
const int maxn = 110;
const int inf = 0x3f3f3f3f;
int maze[maxn][maxn], dis[maxn], n;//maze数组存的是i ,j两条边的权重。
bool visit[maxn];//判断是否走过这个点//走过就标记为true,没走就是false
void Prime() {
	int ans = 0;//设置初始没有边为0,
	visit[1] = true;设置第一个点为走过,
	for(int i = 2; i <= n; i++)
		dis[i] = maze[1][i];//初始dis数组为后面n-1个点到1的最小值,
	for(int i = 1; i < n; i++) {
		int minn = inf ,u = 1;//设置当前最优值,
		for(int j = 1; j <= n; j++)//在mei'you
			if(!visit[j] && dis[j] < minn) {
				minn = dis[j];
				u = j;
			}
		ans += minn;//找到一条最优边,加上他的权重,
		visit[u] = true;
		for(int j = 1; j <= n; j++)//更新当前可以查询的最优边。
			if(!visit[j])
				dis[j] = min(dis[j],maze[u][j]);//比对这条边跟之前已经选过的点是否更好,更好就加入,不是就不加入,
	}
	cout << ans <<endl;
}
int main() {
	cin >> n;
	memset(visit,false,sizeof(visit));//初始化为false,都没有走过,
	for(int i = 1; i <= n; i++)
		for(int j = 1; j <= n; j++)
			cin >> maze[i][j];//输入每条边的权重,
	Prime();
	return 0;
} 
posted @   lifehappy  阅读(401)  评论(0编辑  收藏  举报
点击右上角即可分享
微信分享提示