基于倍增的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;
}
相关推荐
云手机掌柜8 小时前
云手机矩阵:重构企业云办公架构的技术路径与实践落地
智能手机·矩阵·重构
技术与健康11 小时前
LLM实践系列:利用LLM重构数据科学流程01
重构·大模型工程化实践
跨境卫士-小卓1 天前
eBay新政深度解读:2025跨境交易规则重构与卖家应对策略
大数据·重构·跨境电商
聚铭网络2 天前
聚铭安全管家平台2.0实战解码 | 安服篇(四):重构威胁追溯体系
网络·安全·重构
Ashlee_code5 天前
香港券商智能櫃台系統技術解決方案——融合跨境清算與AI風控,助力券商把握滬港雙市爆發機遇**
java·科技·金融·重构·架构·系统架构·php
Jinkxs6 天前
测试工程师的AI转型指南:从工具使用到测试策略重构
人工智能·重构
gptplusplus7 天前
超越“调参”:从系统架构师视角,重构 AI 智能体的设计范式
人工智能·重构·系统架构
轻流AI8 天前
线索转化率翻3倍?AI重构CRM
大数据·人工智能·低代码·重构
weixin_177297220698 天前
剧本杀小程序系统开发:重构推理娱乐生态
小程序·重构·娱乐
思通数据8 天前
AI视频监控:重构安防行业智能化新生态
人工智能·安全·目标检测·机器学习·计算机视觉·重构·数据挖掘