ET算法【最小生成树MST】

一、今日无一小段话

因为在放寒假

---

有很多点和很多条边的权值，问生成最小的包含所有点的图的权值和是多少

二、基于点的prim

P1080 - 【模板】最小生成树 - ETOJ (eriktse.com)

选1号点位起点，for（n - 1）次，每次确定一条边，在所有点中找离intree数组中的点最近的点（d[N]）。加入intree数组，更新所有点到intree数组中任一点的最近距离d[N]。

堆优化O(nlogn)

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;

const int N = 1e5 + 10;
struct edge{
	ll v, w;
	bool operator < (const edge &u) const {
		return w == u.w ? v < u.v : w > u.w;
	}
};
vector<edge> g[N];
ll d[N];
bitset<N> intree;
int main() {
	int n, m; cin >> n >> m;
	memset(d, 0x3f, sizeof d);
	for(int i = 1; i <= m; i++){
		ll u, v, w; cin >> u >> v >> w;
		g[u].push_back({v, w});
		g[v].push_back({u, w});
	}
	ll ans = 0;
	priority_queue<edge> pq;
	pq.push({1, 0});
	d[1] = 0;
	while(pq.size()){
		int x = pq.top().v;
		pq.pop();
		
		if(intree[x]) continue;
		intree[x] = true;
		ans += d[x];
		d[x] = 0;
		
		for(auto &[y, w] : g[x]){
			if(!intree[y] && d[y] > w){
				d[y] = w;
				pq.push({y, d[y]});
			}
		}
	}
	for(int i = 1; i <= n; i++) if(!intree[i]) ans = -1;
	cout << ans <<'\n'; 	
}

三、基于边的kruskal

贪心

(1) 给边从小到大排序

(2) 从小到大选边 {u, v, w}

若 u, v 已连通(并查集) : 跳过    未连通 : 选上, 并连通

O(mlogm) m为边的数量

#include<bits/stdc++.h>
using namespace std;
#define qio ios::sync_with_stdio(0), cin.tie(0), cout.tie(0);
typedef long long ll;
typedef double db;

const int N = 1e5 + 10;
struct edge{
	ll u, v, w;
	bool operator < (const edge &x) const {
		return w == x.w ? (v == x.v ? v < x.v : u < x.u) : w < x.w;
	}
};
int pre[N];
bitset<N> intree;
int root(int x){
	return pre[x] = (pre[x] == x ? x : root(pre[x]));
}
int main() {
	int n, m; cin >> n >> m;
	vector<edge> es;
	for(int i = 1; i <= m; i++){
		int u, v, w; cin >> u >> v >> w;
		es.push_back({u, v, w});
	}
	sort(es.begin(), es.end());
	for(int i = 1; i <= n; i++) pre[i] = i;
	ll ans = 0;
	for(auto &[u, v, w] : es){
		if(root(u) == root(v)) continue;
		ans += w;
		pre[root(u)] = root(v);		
	}
	for(int i = 1; i < n; i++) if(root(i) != root(i + 1)) ans = -1;
	cout << ans << '\n';
}