最小生成树

prim算法

#include<bits/stdc++.h>
using namespace std;
const int maxn = 1e2;
const int INF = 1<<30;//表示无穷大，两个顶点不存在边
int g[maxn][maxn];//利用邻接矩阵表示有权图
int n;//表示顶点个数,从1开始编号

struct edge{
	int v;//边的一个顶点，另一个顶点用数组的索引表示
	int w;//边的权重
};

void init(){
	for(int i=1;i<=n;i++){
		for(int j=1;j<=n;j++)g[i][j]=INF;
	}
}//对图进行初始化


void prim(int v=1){//从v开始构造生成树
	bool reach[maxn]={false};//为true时代表在最小生成树点集中
	edge e[maxn];
	for(int i=1;i<=n;i++){
		e[i].v=v;
		e[i].w=g[v][i];
	}//对边进行初始化
	reach[v]=true;//加入最小生成树点集
	while(true){
		int Min=INF,index=-1;
		for(int i=1;i<=n;i++)if(!reach[i]&&Min>e[i].w)Min=e[i].w,index=i;//寻找待加入最小生成树点集的顶点
		if(index==-1)return ;//顶点已全部加入最小生成树点集中，退出
		cout<<e[index].v<<"-->"<<index<<endl;
		reach[index]=true;//加入最小生成树点集
		for(int i=1;i<=n;i++){
			if(!reach[i]&&g[index][i]<e[i].w){
				e[i].w=g[index][i];
				e[i].v=index;
			}
		}//修改
	}
}

int main(){
	int m;
	int a,b,c;
	scanf("%d%d",&n,&m);
	init();
	for(int i=0;i<m;i++){
		scanf("%d%d%d",&a,&b,&c);
		g[a][b]=g[b][a]=c;
	}
	prim(1);
}

测试输入

7 9
1 2 28
1 6 10
2 3 16
2 7 14
3 4 12
4 5 22
4 7 18
5 6 25
5 7 24 

输出

1-->6
6-->5
5-->4
4-->3
3-->2
2-->7

Kruskal算法

#include<bits/stdc++.h>
using namespace std;
const int maxn = 1e2;
const int INF = 1<<30;//表示无穷大，两个顶点不存在边
int g[maxn][maxn];//利用邻接矩阵表示有权图
int n;//表示顶点个数,从1开始编号
int f[maxn];
struct edge{
	int v1,v2;//边的一个顶点，另一个顶点用数组的索引表示
	int w;//边的权重
	bool operator<(const edge& a) const
    {
        return w > a.w; // 重定义比较函数，使用小根堆
	}
	edge(int a,int b,int c):v1(a),v2(b),w(c){}
};

void init(){
	for(int i=1;i<=n;i++){
		for(int j=1;j<=n;j++)g[i][j]=INF;
	}
}//对图进行初始化

int find(int x){
	return x==f[x]?x:f[x]=find(f[x]);
}//查找共同祖先


void kruskal(int v=1){//从v开始构造生成树
	for(int i=1;i<=n;i++)f[i]=i;//初始化并查集
	priority_queue<edge>p;//构造优先级队列
	for(int i=2;i<=n;i++){
		for(int j=1;j<i;j++){
			int w = g[i][j];
			if(w!=INF)p.push(edge(i,j,w));//依次把每条边加入优先队列
		}
	}
	while(!p.empty()){
		edge e =p.top();
		p.pop();
		int xx=find(e.v1),yy=find(e.v2);
		if(xx==yy)continue;//该两个顶点在一个集合
		cout<<e.v1<<"-->"<<e.v2<<endl;
		f[xx]=yy;//把两个集合合并
	}
}

int main(){
	int m;
	int a,b,c;
	scanf("%d%d",&n,&m);
	init();
	for(int i=0;i<m;i++){
		scanf("%d%d%d",&a,&b,&c);
		g[a][b]=g[b][a]=c;
	}
	kruskal(1);
}

测试输出

6-->1
4-->3
7-->2
3-->2
5-->4
6-->5