ZOJ2838【LCA在线查询】

模板题。

推荐博文：点我打开链接

博主再啰嗦几句：

储存遍历序列的数组ver[ ];

每个结点第一次出现的位置的数组 First[ ];

当前位置的深度的数组 deep[ ]

 

如果要查询u 和v 的LCA

查询 First[ u ] , First[ v ] 遍历区间深度的最小值

具体实现：

1.      DFS一遍，得：

储存遍历的结点：ver[ i ] = x

每个结点第一次遍历到的位置：First[ x ] = index

当前位置的深度 deep[ i ] = dep

 

2.      建立ST表，区间RMQ

通过 First 数组访问到 Left, Right 位置

通过 deep 数组维护 dp[ i ][ j (len= (1<<j)) ] 区间最小值的位置

区间RMQ 求得k

While(1<<(k+1)<=Right-Left+1) k++;

得区间 dp[ Left ][ k ]， dp[ Right – (1<<k) + 1 ][ k ];

然后比较大小可得位置，通过 ver 数组返回结点( LCA ).

题意：

 Then for each query 'A B C',
 if village C is on the path between village A and B,
    output 'Yes' (without quotation) in a single line,
    otherwise output 'No' (without quotation).

思路：

询问两点之间是不是经过某个点，

AB的LCA==C是AB的LCA；

C是A的根，CA的LCA==AB的LCA

C是B的根，CB的LCA==AB的LCA

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;

const int N = 5e4 + 10;
const int M = 1e5 + 10;
struct Edge{
    int to;
    int next;
} e[M];
int head[N],tol,n;
void init()
{
    tol=0;
    memset(head,-1,sizeof(head));
}
void add(int u,int v)
{
    e[tol].to=v;
    e[tol].next=head[u];
    head[u]=tol++;
}

bool vis[N];
int deep[N<<2],first[N],tot,id[N<<2];
void DFS(int u,int dep)
{
    vis[u]=true;
    id[++tot]=u;
    deep[tot]=dep;
    first[u]=tot;
    for(int i=head[u]; ~i; i=e[i].next){
        int v=e[i].to;
        if(vis[v]) continue;
        DFS(v,dep+1);
        id[++tot]=u;
        deep[tot]=dep;
    }
}
int dp[N<<2][35];
void ST()
{
    for(int i=1; i<=tot; i++) dp[i][0]=i;
    for(int j=1; (1<<j)<=tot; j++){
        for(int i=1; i+(1<<j)-1<=tot; i++){
            int a=dp[i][j-1],b=dp[i+(1<<(j-1))][j-1];
            dp[i][j]=deep[a] < deep[b]?a:b;
        }
    }
}

int RMQ(int Left,int Right)
{
    int k=0;
    while((1<<(k+1)) <= (Right-Left+1)) k++;
    int a=dp[Left][k],b=dp[Right-(1<<k)+1][k];
    return deep[a]<deep[b]?a:b;
}

int LCA(int u,int v)
{
    int x=first[u],y=first[v];
    if(x>y) swap(x,y);
    int res=RMQ(x,y);
    return id[res];
}

int main()
{
    int cas=1;
    int u,v,m,c;
    while(~scanf("%d",&n))
    {
        init();
        if(cas>1) puts("");
        printf("Case %d:\n",cas++);
        for(int i=0; i<n-1; i++)
        {
            scanf("%d%d",&u,&v);
            add(u,v);
            add(v,u);
        }
        tot=0;
        memset(vis,false,sizeof(vis));
        DFS(0,1);
        ST();
        
        scanf("%d",&m);
        while(m--)
        {
            scanf("%d%d%d",&u,&v,&c);
            int uc=LCA(u,c);
            int vc=LCA(v,c);
            int uv=LCA(u,v);
            if(uc == c && vc == c && uv == c) puts("Yes");
            else if((uc == c && vc == v) || (uc == u && vc == c)) puts("Yes");
            else if((uc == c && uv == vc) || (uc == uv && vc == c)) puts("Yes");
            else puts("No");
        }
    }
    return 0;
}