最小生成树Prim和Kruskal

最新推荐文章于 2022-07-05 07:30:00 发布

ertcmmy

最新推荐文章于 2022-07-05 07:30:00 发布

阅读量87

点赞数

分类专栏：算法

本文链接：https://blog.csdn.net/ertcmmy/article/details/96730792

版权

算法专栏收录该内容

9 篇文章 0 订阅

订阅专栏

Kruskal

int n, m; // 顶点和边数
int u[100], v[100], w[100]; // 边顶点和权重
int r[100]; // 边权重从小到大的下标
int pa[100]; // 并查集所用

void read(){
	cin >> n >> m;
	for(int i = 0; i < m; i++){
		int x, y, z;
		cin >> x >> y >> z;
		u[i] = x;
		v[i] = y;
		w[i] = z;
	}
}

// 查找u的根节点（所属集合）
int find(int u){
	return u == pa[u] ? u : pa[u] = find(pa[u]);
}

bool cmp(const int &a, const int &b){
	return w[a] < w[b];
}

int kruskal(){
	// 初始化r
	for(int i = 0; i < m; i++){
		r[i] = i;
	}
	sort(r, r + m, cmp); // 按给定比较函数排序

	// 初始化并查集
	for(int i = 0; i < n; i++){
		pa[i] = i;
	}
	
	int ans = 0; // 最小生成树的权重
	for(int e = 0; e < m; e++){
		int x = u[e], y = v[e], z = w[e];
		int p1 = find(x), p2 = find(y);
		if(p1 != p2){
			ans += z;
			pa[p1] = p2;
		}
	}

	return ans;
}