【洛谷】3379 【模板】最近公共祖先(LCA)

22 篇文章 0 订阅
10 篇文章 0 订阅

题目传送门

诶,自己真的是越来越菜了,自己寒假里学会的LCA到现在已经忘得差不多了。

今天就当是复习一下吧。

其实这个倍增的思想还是挺神奇的,每次询问时先把深度较大的节点倍增到和深度较小的节点同一深度。

然后再把这两个节点同时倍增到离它们最远的不相同的祖先出,赋值并循环,最后返回当前任意一个节点的父亲节点就是这两个节点的LCA了。

附上AC代码:

#include <cstdio>
#include <cctype>
#include <cmath>
#include <algorithm>
#define N 500010
using namespace std;

struct side{
	int to,nt;
}s[N<<1];
int n,m,st,x,y,num,h[N],d[N],f[N][21],k;
bool b[N];

inline char nc(){
	static char ch[100010],*p1=ch,*p2=ch;
	return p1==p2&&(p2=(p1=ch)+fread(ch,1,100010,stdin),p1==p2)?EOF:*p1++;
}

inline void read(int &a){
	static char c=nc();int f=1;
	for (;!isdigit(c);c=nc()) if (c=='-') f=-1;
	for (a=0;isdigit(c);a=a*10+c-'0',c=nc());
	a*=f;return; 
}

inline void add(int x,int y){
	s[++num]=(side){y,h[x]},h[x]=num;
	s[++num]=(side){x,h[y]},h[y]=num;
}

inline void build(int x){
	b[x]=1;
	for (int i=h[x]; i; i=s[i].nt)
		if (!b[s[i].to])
			d[s[i].to]=d[x]+1,f[s[i].to][0]=x,build(s[i].to);
	return;
}

inline int query(int x,int y){
	if (d[x]<d[y]) swap(x,y);
	for (int i=k; i>=0; --i)
		if (d[f[x][i]]>=d[y]) x=f[x][i];
	if (x==y) return x;
	for (int i=k; i>=0; --i)
		if (f[x][i]!=f[y][i]) x=f[x][i],y=f[y][i];
	return f[x][0];
}

int main(void){
	read(n),read(m),read(st);
	for (int i=1; i<n; ++i) read(x),read(y),add(x,y);
	d[st]=1,build(st),k=log(n)/log(2)+1;
	for (int j=1; j<=k; ++j)
		for (int i=1; i<=n; ++i)
			f[i][j]=f[f[i][j-1]][j-1];
	for (int i=1; i<=m; ++i){
		read(x),read(y);
		printf("%d\n",query(x,y));
	}
	return 0;
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值