Prim 模版

版权声明:希望能帮到弱校的ACMer成长,因为自己是弱校菜鸡~~~~ https://blog.csdn.net/Mr__Charles/article/details/82838158
/*选点开始
  分为已加入树点集U和未加入树点集V
  类似于最短路径,但是,是指点加入U的最短距离 
*/ 
#include<cstdio>
#include<iostream>
#include<algorithm>
#define maxn 55
#define INF 0x3f3f3f3f
using namespace std;

int n,m,sum,dis[maxn];
int maps[maxn][maxn];
bool vis[maxn];

void init(){
	sum = 0;
	for(int i = 1; i <= n ; i++){
		dis[i] = INF;
		vis[i] = false;
		for(int j = 1; j <= i; j++)
		    i == j ? maps[i][j] = 0 : maps[i][j] = maps[j][i] = INF;
	}
}

void Prim(int s){
	dis[s] = 0;
	for(int i = 1; i <= n ; i++){
		int p = 0, minn = INF;
		for(int j = 1; j <= n; j++){
			if(!vis[j] && minn > dis[j]){
				p = j;
				minn = dis[j];
			}
		}
		vis[p] = true;
		sum += dis[p];
		for(int j = 1; j <= n ; j++){
			if(!vis[j] && dis[j] > maps[p][j])
			    dis[j] = maps[p][j];
		}
	}
}

猜你喜欢

转载自blog.csdn.net/Mr__Charles/article/details/82838158