算法提高模板LCA

模板:

cpp 复制代码
#include<bits/stdc++.h>

using namespace std;

typedef long long ll;
const int MOD = 998244353;
const int N = 2e5 + 10;

int n, m, s, a, b;
vector<int>e[N];
int dep[N], fa[N][20];//深度
//第u个节点向上走2的i次方之后所对应的祖先
void dfs(int u, int father)
{
	dep[u] = dep[father] + 1;
	fa[u][0] = father;//初始化递归的第一项
	for(int i = 1; i <= 19; i ++)
	{
		fa[u][i] = fa[fa[u][i - 1]][i - 1];
	}
	for(int v : e[u])
	{
		if(v != father) dfs(v, u);
	}
}

int lca(int u, int v)
{
	if(dep[u] < dep[v]) swap(u, v);
	//先跳到同一层
	for(int i = 19; i >= 0; i --)
	{
		if(dep[fa[u][i]] >= dep[v])
		{
			u = fa[u][i];
		}//一步一步的跳
	}
	if(u == v) return v;
	for(int i = 19; i >= 0; i --)
	{
		if(fa[u][i] != fa[v][i])
		{
			u = fa[u][i], v = fa[v][i];
		}
	}
	return fa[u][0];
}
int main()
{
	cin >> n;
	while(n --){
		cin >> a >> b;
		if(b == -1) s = a;
		else {
			e[a].push_back(b);
			e[b].push_back(a);
		}
		
	}
	dfs(s, 0);
	cin >> m;
	
	while(m --){
		cin >> a >> b;
		int o = lca(a, b);
		
		if(o == a) cout << 1 << endl;
		else if(o == b) cout << 2 << endl;
		else cout << 0 << endl;
	}
	return 0;
}
相关推荐
用户50408278583920 分钟前
1. RAG 权威指南:从本地实现到生产级优化的全面实践
算法
Python×CATIA工业智造1 小时前
详细页智能解析算法:洞悉海量页面数据的核心技术
爬虫·算法·pycharm
无聊的小坏坏2 小时前
力扣 239 题:滑动窗口最大值的两种高效解法
c++·算法·leetcode
黎明smaly2 小时前
【排序】插入排序
c语言·开发语言·数据结构·c++·算法·排序算法
YuTaoShao3 小时前
【LeetCode 热题 100】206. 反转链表——(解法一)值翻转
算法·leetcode·链表
YuTaoShao3 小时前
【LeetCode 热题 100】142. 环形链表 II——快慢指针
java·算法·leetcode·链表
CCF_NOI.3 小时前
(普及−)B3629 吃冰棍——二分/模拟
数据结构·c++·算法
运器1234 小时前
【一起来学AI大模型】支持向量机(SVM):核心算法深度解析
大数据·人工智能·算法·机器学习·支持向量机·ai·ai编程
Zedthm4 小时前
LeetCode1004. 最大连续1的个数 III
java·算法·leetcode
lcg_magic4 小时前
图论系列(一):基础概念与术语解析
图论