LCA--最近公共祖先模板

1 篇文章 0 订阅
#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int N=1e4;
struct Node{
    int v,next;
}edge[N<< 1]; 
int head[N],cnt;

void add_edge(int u, int v) 
{
    edge[cnt].v=v;
    edge[cnt].next=head[u];
    head[u]=cnt++;
}

int depth[N], fa[N][22], lg[N];

void dfs(int now, int fath) {
    fa[now][0] = fath; 
	depth[now] = depth[fath] + 1;
	
    for(int i = 1; i <= lg[depth[now]]; ++i)
        fa[now][i] = fa[fa[now][i-1]][i-1];
    
	for(int i = head[now]; i!=-1; i = edge[i].next)
        if(edge[i].v != fath) dfs(edge[i].v, now);
}

int LCA(int x, int y) {
    if(depth[x] < depth[y]) 
	swap(x, y);
	
    while(depth[x] > depth[y])
        x = fa[x][lg[depth[x]-depth[y]] - 1];
        
    if(x == y) return x;
    
    for(int k = lg[depth[x]] - 1; k >= 0; --k)
        if(fa[x][k] != fa[y][k])
            x = fa[x][k], y = fa[y][k];
    
    return fa[x][0];
}
int main() {
	//树的节点数目为n,查询次数为m,根节点为s 
    int n, m, s;
	scanf("%d%d%d", &n,&m,&s);
	memset(head,-1,sizeof(head));
	cnt=0;
	for(int i=1;i<n;i++)
	{
		int u,v;
		scanf("%d%d",&u,&v);
		add_edge(u,v);
		add_edge(v,u); 
	}
	
    for(int i = 1; i <= n; ++i)
        lg[i] = lg[i-1] + (1 << lg[i-1] == i);
    dfs(s, 0);
    
    for(int i = 1; i <= m; ++i) 
	{
        int x, y; scanf("%d%d",&x, &y);
        printf("%d\n", LCA(x, y));
    }
    return 0;
}

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值