AcWing859 kruskal算法求最短路

2022/1/30 14:04:55

本文主要是介绍AcWing859 kruskal算法求最短路,对大家解决编程问题具有一定的参考价值,需要的程序猿们随着小编来一起学习吧!

#include<iostream>
#include<algorithm>
using namespace std;
const int N = 2e5 + 10;
int n, m;
int p[N];

struct edge {
	int a, b, w;
	bool operator< (const edge& W)const
	{
		return w < W.w;
	}
};
edge edges[N];

int find(int x) {
	if (x != p[x])p[x] = find(p[x]);
	return p[x];
}

int kruskal() {
	int cnt = 0, res = 0;//cnt统计边数 res统计权重和
	sort(edges, edges + m);
	for (int i = 0; i < m; i++) {
		int a = edges[i].a, b = edges[i].b, w = edges[i].w;
		a = find(a), b = find(b);
		if (a != b) {
		//如果不在一个集合中
			p[a] = b;
			res += w;
			cnt++;
		}
	}
	if (cnt < n - 1)return 0x3f3f3f3f;
	else return res;
}
int main() {
	cin >> n >> m;
	for (int i = 0; i < n; i++)p[i] = i;
	for (int i = 0; i < m; i++) {
		int a, b, w;
		cin >> a >> b >> w;
		edges[i] = { a,b,w };

	}
	int t = kruskal();
	if (t == 0x3f3f3f3f)puts("impossible");
	else cout << t << endl;
}


这篇关于AcWing859 kruskal算法求最短路的文章就介绍到这儿,希望我们推荐的文章对大家有所帮助,也希望大家多多支持为之网!


扫一扫关注最新编程教程