lca板子

#include<bits/stdc++.h>
using namespace std;
const int N=5e5+10;
int head[N];
int n,m,s,f[N][30],lg[N],h[N];
///该结点的深度
///f[i][j]为i结点向上2^j的祖先
struct node
{
    int to,ne;
}bian[N*2];
int tot = 0;
void add_edge(int x,int y)
{
   tot++;
   bian[tot].to = y;
   bian[tot].ne = head[x];
   head[x] = tot;
}

void dfs(int x,int fx)///一个结点和他的父亲结点
{
    h[x] = h[fx]+1;///深度就加一
    f[x][0] = fx;

    for(int i=1; ( 1<<i )<=h[x]; i++)
    {
        f[x][i]=f[ f[x][i-1] ][i-1];
    }

    for(int i=head[x];i;i=bian[i].ne)
    {
        if(bian[i].to!=fx)///防止走回去了
        {
            dfs(bian[i].to,x);
        }
    }
}

int LCA(int x,int y)
{
    if(h[x]<h[y])
        swap(x,y);

    while(h[x]>h[y])
        x=f[x][ lg[h[x]-h[y]] ];

    if(x==y)
        return x;
    for(int i=lg[h[x]]; i>=0; i--)
        if(f[x][i]!=f[y][i])
            x=f[x][i],y=f[x][i];
    return f[x][0];
}

int main()
{
    scanf("%d %d %d",&n,&m,&s);

    for(int i=1; i<n; i++)
    {
        int x,y;
        scanf("%d %d",&x,&y);
        add_edge(x,y);
        add_edge(y,x);
    }

    dfs(s,0);

    for(int i=2; i<=n; i++)
        lg[i]=lg[i>>1]+1;

    while(m--)
    {
        int a,b;
        scanf("%d %d",&a,&b);
        printf("%d\n",LCA(a,b));
    }
    return 0;
}

上一篇:makefile "=", ":=", "?="


下一篇:谈谈实现Serializable接口的作用和必要性