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

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

一、视频讲解

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

二、正解代码

cpp 复制代码
//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();
}
相关推荐
神探阿航4 小时前
第十五届蓝桥杯大赛软件赛省赛C/C++ 大学 B 组
java·算法·蓝桥杯
不能只会打代码6 小时前
蓝桥杯例题一
算法·蓝桥杯
ExRoc8 小时前
蓝桥杯真题 - 填充 - 题解
c++·算法·蓝桥杯
hero_heart9 小时前
重构(1)if-else
重构
AI量化投资实验室9 小时前
deap系统重构,再新增一个新的因子,年化39.1%,卡玛提升至2.76(附python代码)
大数据·人工智能·重构
BingLin-Liu10 小时前
蓝桥杯备考:红黑树与map和set
职场和发展·蓝桥杯
L73S3710 小时前
C++入门(1)
c++·程序人生·考研·蓝桥杯·学习方法
2401_8979168413 小时前
2018 秋招 百度二轮面试---血淋淋的经历写实
面试·职场和发展
醇醛酸醚酮酯16 小时前
Leetcode热题——移动零
算法·leetcode·职场和发展
qy发大财16 小时前
平衡二叉树(力扣110)
数据结构·算法·leetcode·职场和发展