RMQ求LCA

落谷模板

#include<iostream>#include<cstdio>#include<algorithm>#include<cstring>#include<string>#define MAXN 1010001using 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));    }}