基于倍增的LCA + kruskal重构树 + 并查集

P2245 星际导航

cpp 复制代码
//基于倍增的LCA + kruskal重构树 + 并查集
#include<bits/stdc++.h>
#define endl '\n'
#define rep(_start_,_end_) for(int i =  _start_;i <= (_end_);i ++)
#define all0(x) (x).begin(),(x).end()
#define all1(x) (x).begin() + 1,(x).end()
using namespace std;
using ll = long long;

const int N = 1e5 + 10; // 原图最大顶点数
const int M = 5e5 + 10; // 原图最大边数

int n, m, q;
int par[N * 2];
int val[N * 2];//非叶子的权值
vector<int>trie[N * 2];


struct EDGE
{
	int u, v, w;
};

int Find(int x)
{
	return (par[x] == x) ? x : par[x] = Find(par[x]);
}
int new_node;
void build_kruskal(vector<EDGE>&edge)
{
	//初始化
	for(int i = 1;i <= 2 * n - 1;i ++)
		par[i] = i;
	//遍历所有边
	new_node = n;
	for(auto &[l,r,w] : edge)
	{
		int rootl = Find(l);
		int rootr = Find(r);
		
		if(rootl != rootr)
		{
			new_node ++;
			val[new_node] = w;
//			cout << "newnode: "<<new_node <<endl;
			trie[new_node].push_back(rootl);
			trie[new_node].push_back(rootr);
			
			par[rootl] = new_node;
			par[rootr] = new_node;
		}
	}
}


int dep[N * 2],fa[N * 2][20];

void dfs(int u,int u_father)
{

	dep[u] = dep[u_father] + 1;
	fa[u][0] = u_father;
	
	for(int j = 1;j <= 19;j ++)
	{
		fa[u][j] = fa[fa[u][j - 1]][j - 1];
	}
	
	for(auto v:trie[u])
	{
		if(v != u_father)
		{
			dfs(v,u);
		}
	}
}


int get_lca(int l,int r)
{
	if(dep[l] < dep[r])swap(l,r);
	for(int i = 19;i >= 0;i --)
	{
		if(dep[fa[l][i]] >= dep[r])
			l = fa[l][i];
	}
	
	if(l == r)return l;
	
	for(int i = 19;i >= 0;i --)
	{
		if(fa[l][i] != fa[r][i])
			l = fa[l][i],r = fa[r][i];
	}
	return fa[l][0];
}

void solve()
{
	cin >> n >> m; 
	vector<EDGE> edge(m);
	rep(0, m - 1) cin >> edge[i].u >> edge[i].v >> edge[i].w;
	
	sort(all0(edge), [](const auto&i, const auto&j){
		return i.w < j.w;
	});
	//重构树
	build_kruskal(edge);
	//预处理dfs
	for(int i = 1;i <= new_node ;i ++)
	{
		if(par[i] == i)
		{
			dfs(i,0);
		}
	}
//	for(int i = 1;i <= 2 * n - 1;i ++)cout << fa[i][0]<<" ";
//	cout << endl;
	cin >> q;
	while(q --)
	{
		int l,r;
		cin >> l >> r;
		if(Find(l) != Find(r))
			cout << "impossible"<<endl;
		else{
			cout << val[get_lca(l,r)] <<endl;
		}
	}
}

int main()
{
	ios_base::sync_with_stdio(false);
	cin.tie(NULL);
	int t;
	// cin >> t;
	t = 1;
	
	while(t --)
	{
		solve();
	}
	return 0;
}
相关推荐
区块block27 分钟前
加密市场再添新势力,BUYCOIN生态代币BCT以多维赋能重构交易所生态价值
重构·区块链
b***251129 分钟前
气动点焊机如何重构动力电池制造逻辑
重构·制造
Alter123033 分钟前
用AI重构人机关系,OPPO智慧服务带来了更“懂你”的体验
人工智能·重构
GIS数据转换器3 小时前
带高度多边形,生成3D建筑模型,支持多种颜色或纹理的OBJ、GLTF、3DTiles格式
数据库·人工智能·机器学习·3d·重构·无人机
专注VB编程开发20年2 天前
产品更新与重构策略:创新与稳定的平衡之道
重构·erp·进销存
hqyjzsb2 天前
2025年市场岗位能力重构与跨领域转型路径分析
c语言·人工智能·信息可视化·重构·媒体·改行学it·caie
rengang662 天前
智能化的重构建议:大模型分析代码结构,提出可读性和性能优化建议
人工智能·性能优化·重构·ai编程
一尘之中2 天前
觉醒的拓扑学:在量子纠缠与神经幻象中重构现实认知
人工智能·重构
Guheyunyi2 天前
风险感知中枢:监测预警系统的架构与核心
大数据·运维·安全·重构·架构·自动化
CodeJourney.2 天前
AI产业技术突破、生态重构与场景深耕
人工智能·重构