hdu 5266 pog loves szh III LCA+RMQ

题意:

给你一棵树,然后询问l~r节点的最近公共祖先(LCA)。

思路:

用RMQ维护一段区间的LCA,然后询问时,将两个区间的LCA再求一次LCA即可。

code:

#pragma comment(linker, "/STACK:102400000,102400000")
#include <cstdlib>
#include <cstdio>
#include <cstring>
#include <iostream>
using namespace std;

const int N = 3e5+5;
typedef long long LL;

int n, q;
int head[N], cnt;
struct Edge {
    int v, next;
}edge[N<<1];

int par[N][30];
int dep[N];
int d[N];
int rmq[N][30];

void init() {
    memset(head, -1, sizeof(head));
    cnt = 0;
}

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

void dfs(int u, int p, int d) {
    dep[u] = d; par[u][0] = p;
    for(int i = head[u];i != -1; i = edge[i].next) {
        int v = edge[i].v;
        if(v == p) continue;
        dfs(v, u, d+1);
    }
}
    
int lca(int u, int v) {
    if(dep[u] > dep[v]) swap(u, v);
    for(int i = 0;i < 20; i++) {
        if(((dep[v]-dep[u])>>i) & 1)
            v = par[v][i];
    }
    if(u == v) return v;
    for(int k = 20;k >= 0; k--)
        if(par[v][k] != par[u][k]) {
            u = par[u][k];
            v = par[v][k];
        }
    return par[u][0];
}

int query(int l, int r) {
    int len = r-l+1;
    int k = 0;
    while((1<<(k+1)) <= len) k++;
    return lca(rmq[l][k], rmq[r-(1<<k)+1][k]);
}
    
int main() {
    while(scanf("%d", &n) != EOF) {
        init();
        int u, v;
        for(int i = 0;i < n-1; i++) {
            scanf("%d%d", &u, &v);
            addEdge(u, v);
            addEdge(v, u);
        }

        dfs(1, -1, 0);
        for(int i = 0;i+1<20; i++) {
            for(int u = 1;u <= n; u++) {
                if(par[u][i] < 0) par[u][i+1] = -1;
                else par[u][i+1] = par[par[u][i]][i];
            }
        }
        for(int i = 1;i <= n; i++) rmq[i][0] = i;
        for(int k = 0;k < 20; k++) 
            for(int i = 1;i <= n; i++) {
                if(i+(1<<(k+1))-1 > n) break;
                rmq[i][k+1] = lca(rmq[i][k], rmq[i+(1<<k)][k]);
                //cout<<rmq[i][k+1]<<endl;
            }
        //cout<<rmq[2][2]<<endl;
        scanf("%d", &q);
        int l, r;
        while(q--) {
            scanf("%d%d", &l, &r);
            printf("%d\n", query(l, r));
        }
    }
    return 0;
}



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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值