蓝桥杯真题讲解:网络稳定性(Kruskal重构树+LCA)

蓝桥杯真题讲解:网络稳定性(Kruskal重构树+LCA)

一、视频讲解

蓝桥杯真题讲解:网络稳定性(Kruskal重构树+LCA)
在这里插入图片描述

二、正解代码

//kruskal重构树 + lca
#include<bits/stdc++.h>
#define endl "\n"
#define deb(x) cout << #x << " = " << x << '\n';
#define INF 0x3f3f3f3f
using namespace std;
const int N = 2e5 + 10;
const int M = 3e5 + 10;
int n, m, q;

struct edge{
	int x, y, w;
}e[M];
vector<int>g[N];
int p[N];
int val[N];//重构树节点的权值。
int find(int x){
	if(x != p[x])p[x] = find(p[x]);
	return p[x];
}
void kruskal(){
	for(int i = 1; i <= n * 2; i ++){
		p[i] = i;
	}
	sort(e, e + m, [&](edge x, edge y){return x. w > y.w;});
	int id = n;
	for(int i = 0; i < m; i ++){
		int x = e[i].x, y = e[i].y, w = e[i].w;
		
		int px = find(x), py = find(y);
		if(py == px)continue;
		id ++;
		p[px] = id, p[py] = id;
		g[id].push_back(py);
		g[id].push_back(px);
		val[id] = w;
	}
}

int dep[N], fa[N], siz[N];
int top[N], hs[N];
void dfs1(int u, int f){
	dep[u] = dep[f] + 1;
	siz[u] = 1;
	fa[u] = f;
	for(auto s: g[u]){
		if(s == f)continue;
		dfs1(s, u);
		siz[u] += siz[s];
		if(siz[s] > siz[hs[u]])
			hs[u] = s;
	}
}

void dfs2(int u, int t) {
	top[u] = t;
	if(!hs[u])return;
	dfs2(hs[u], t);
	for(auto s: g[u]){
		if(s == fa[u] || s == hs[u])continue;
		dfs2(s, s);
	}
}

int lca(int u, int v){
	while(top[u] != top[v]){
		if(dep[top[u]] < dep[top[v]])
			swap(u, v);
		u = fa[top[u]];
	}
	return dep[u] < dep[v] ? u : v;
}

void solve()
{
	cin >> n >> m >> q;
	for(int i = 0; i < m; i ++){
		cin >> e[i].x >> e[i].y >> e[i].w;
	}
	kruskal();
	for(int i = 1; i <= 2 * n; i ++){
		if(p[i] == i){
			//该连通块中lca的预处理
			dfs1(i, 0);
			dfs2(i, i);
		}
	}

	while(q --){
		int a, b; cin >> a >> b;
		if(find(a) != find(b)){
			cout << -1 << endl;
			continue;
		}
		cout << val[lca(a, b)] << endl;
	}

}

signed main()
{
	ios::sync_with_stdio(0);
	cin.tie(0);
	cout.tie(0);
	int t = 1;
	//cin >> t;
	while(t--)
	solve();
}
  • 8
    点赞
  • 7
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值