RMQ求LCA

落谷模板

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<string>
#define MAXN 1010001
using namespace std;
int n,m,to[MAXN],nxt[MAXN],s,head[MAXN],e,pos[MAXN],dfn[MAXN],u,v,dep[MAXN],lg[MAXN];
int st[MAXN][20];
inline void add( int x,int y)
{
    e++;
    to[e]=y;
    nxt[e]=head[x];
    head[x]=e;
}
void dfs(int x,int f)
{
    dfn[++e]=x;
    pos[x]=e;
    for(int i=head[x];i;i=nxt[i])
    {
        int y=to[i];
        if(y!=f)
        {
            dep[y]=dep[x]+1;
            dfs(y,x);
            dfn[++e]=x;
        }
    }


}
int lca(int x,int y)
{
    int pc,pcc,l;
    if(pos[x]>pos[y]) swap(x,y);
    pc=pos[x];
    pcc=pos[y];
    l=lg[pcc-pc+1];
    if(dep[st[pc][l]]<dep[st[pcc-(1<<l)+1][l]])
        return st[pc][l];
        else
        return st[pcc-(1<<l)+1][l];
}
int main()
{
//  freopen("1.txt","r",stdin);
    cin>>n>>m>>s;
    memset(dep,0x7f,sizeof(dep));
    for(int i=1;i<n;i++)
    {
        scanf("%d%d",&u,&v);
        add(u,v);
        add(v,u);
    }
    e=0;
    dep[s]=0;
    dfs(s,0);
    lg[0]=-1;
    for(int i=1;i<=e;i++)
    {
        st[i][0]=dfn[i];
    }
    //i<=n就行了
    for(int i=1;i<=e;i++)
    {
        if((i&(i-1))==0) lg[i]=lg[i-1]+1;
        else lg[i]=lg[i-1];
    }
    for(int i=1;i<=lg[e];i++)   //j+(1<<i)-1<=e就行了
    {
        for(int j=1;j+(1<<i)-1<=e;j++)
        {
            //右边的dep[st[j+(1<<(i-1)][i-1]]!!!
            //                ↑
            //下面else里也是
            if(dep[st[j][i-1]]<dep[st[j+(1<<(i-1))][i-1]])
            st[j][i]=st[j][i-1];
            else st[j][i]=st[j+(1<<(i-1))][i-1];
        }   
    }
    for(int i=1;i<=m;i++)
    {
        scanf("%d%d",&u,&v);
        printf("%d\n",lca(u,v));
    }
}