2021-02-25 最小生成树模板——krusal算法

摘要:

最小生成树算法模板——krusal算法


题目简述:

洛谷P3366最小生成树模板


算法分析:

krusal算法的主要利用了贪心的思想,现将所有边按照权重排序,之后每一次添加一条权值最小的边。由于初始时将图简化为只含有n个点的平凡图,因此新加入的边满足加入后不产生环路。
初始时n个点,每一次加入一条边都会将两个点相连接,这个特性正好满足并查集的特点。因此,krusal的代码实现可以采用并查集算法。


代码以及详细注释:

#include <iostream>
#include <stdio.h>
#include <vector>
#include <algorithm>
#include <queue>
#include <numeric>
#define INF 10000000
#pragma warning(disable:4996)
using namespace std;

struct edge {
    
    
	int u; 
	int v;
	int w;
};

bool comp(edge& e1, edge& e2)
{
    
    
	return e1.w < e2.w;
}

class UnionFind {
    
    
public:
	int count; //种类数
	vector<int> root; //保存每一个节点的根节点
	UnionFind(int _count = 1) :count(_count)
	{
    
    
		root.resize(_count + 1);
		for (int i = 1; i <= _count; ++i)
			root[i] = i;
	}

	//路径压缩
	int find(int x) {
    
    
		return root[x] == x ? root[x] : root[x] = find(root[x]);
	}

	void Union(int x,int y) {
    
    
		int rootx = find(x);
		int rooty = find(y);
		if (rootx != rooty)
			root[y] = rootx;
	}
};

class Solution {
    
    
public:
	int cnt = 0;
	vector<edge> e;
	int n, m;
	int count = 0;
	int ans = 0;
	inline void add_edge(int u, int v, int w) {
    
    
		cnt++;
		e[cnt].u = u;
		e[cnt].v = v;
		e[cnt].w = w;
	}

	void krusal() {
    
    
		cin >> n >> m;
		e.resize(m + 1);

		for (int i = 0; i < m; ++i)
		{
    
    
			int u, v, w;
			cin >> u >> v >> w;
			add_edge(u, v, w);
		}
		sort(e.begin(), e.end(), comp);
		UnionFind un(n);
		for (int i = 1; i <= m; ++i)
		{
    
    
			int u = un.find(e[i].u);
			int v = un.find(e[i].v);
			if (u == v)
				continue;
			ans += e[i].w;
			count++;
			un.Union(u, v);
		}
		if (count == n - 1)
			cout << ans << endl;
		else
			cout << "orz" << endl;
	}

};

int main() {
    
    

	//freopen("in.txt", "r", stdin);
	Solution s;
	s.krusal();
	return 0;

}

猜你喜欢

转载自blog.csdn.net/sddxszl/article/details/114055589