重链剖分求LCA

#include <bits/stdc++.h>

using namespace std;
#define ll long long
#define mem(a, b) memset(a,b,sizeof(a))
#define INF 0x3f3f3f3f
#define DBG printf("this is a input\n")
#define fi first
#define se second
#define mk(a, b) make_pair(a,b)
#define p_queue priority_queue

ll gcd(ll a, ll b) {
    return b == 0 ? a : gcd(b, a % b);
}

ll lcm(ll a, ll b) {
    return a / gcd(a, b) * b;
}
const int maxn = 500005;
int head[maxn] , cnt = 0;
struct e
{
    int t, next;
}edge[maxn*2];
void add(int f, int t)
{
    edge[cnt].t = t;
    edge[cnt].next = head[f];
    head[f] = cnt ++;
}
int n , m, rt;
int son[maxn] , size[maxn], top[maxn], fa[maxn] , deep[maxn];
map<pair<int,int>,int>ma;
void dfs1(int u)
{
    size[u] = 1;
    for(int i = head[u] ; i != -1 ; i = edge[i].next)
    {
        int v = edge[i].t;
        if(v != fa[u])
        {
            fa[v] = u;
            deep[v] = deep[u] + 1;
            dfs1(v);
            size[u] += size[v];
            if(size[v] > size[son[u]])
                son[u] = v;
        }
    }
}
void dfs2(int u, int t)
{
    top[u] = t;
    if(son[u])
        dfs2(son[u],t);
    for(int i = head[u] ; i != -1; i = edge[i].next)
    {
        int v = edge[i].t;
        if(v != son[u] && v != fa[u])
            dfs2(v,v);
    }
}
int lca(int x , int y)
{
    while(top[x] != top[y])
    {
        if(deep[top[x]] < deep[top[y]])
            swap(x,y);
        x = fa[top[x]];
    }
    return deep[x] > deep[y] ? y : x;
}
int main(void)
{
    mem(head,-1);
    scanf("%d %d %d",&n, &m, &rt);
    for(int i = 1 ; i <= n - 1 ; i ++)
    {
        int u , v;
        scanf("%d %d",&u, &v);
        add(u,v);
        add(v,u);
    }
    dfs1(rt);
    dfs2(rt,rt);
    for(int i = 1 ; i <= m ; i ++)
    {
        int x , y;
        scanf("%d %d",&x, &y);
        printf("%d\n",lca(x,y));
    }
}

 

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

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值