祖孙询问

祖孙询问

Time Limit: 1 Sec   Memory Limit: 128 MB
Submit: 30   Solved: 24
[ Submit][ Status][ Web Board]

Description

已知一棵n个节点的有根树。有m个询问。每个询问给出了一对节点的编号x和y，询问x与y的祖孙关系。

Input

输入第一行包括一个整数n表示节点个数。
接下来n行每行一对整数对a和b表示a和b之间有连边。如果b是-1，那么a就是树的根。
第n+2行是一个整数m表示询问个数。
接下来m行，每行两个正整数x和y。
n,m≤40000，每个节点的编号都不超过40000

Output

对于每一个询问，输出1:如果x是y的祖先，输出2:如果y是x的祖先，否则输出0。

Sample Input

10
234 -1
12 234
13 234
14 234
15 234
16 234
17 234
18 234
19 234
233 19
5
234 233
233 12
233 13
233 15
233 19

Sample Output

1
0
0
0
2

题解：还是练模板啊，这是练LCA的最后一篇了，练到这就差不多了。

结合代码理解吧

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<iostream>
#include<algorithm>
using namespace std;
int n,m,cnt,head[40002],root;
struct Edge{int to,next;} edge[80002];
int fa[40002][20],deep[40002];//2^16=65536
void add(int x,int y)
{
    edge[++cnt].to=y;
    edge[cnt].next=head[x];
    head[x]=cnt;
}
void getf(int x)
{
    for(int i=1;i<=15;i++)
    {
        if(deep[x]<(1<<i))break;
        fa[x][i]=fa[fa[x][i-1]][i-1];
    }
}
void dfs(int x)
{
    getf(x);
    for(int i=head[x];i;i=edge[i].next)
    {
        if(edge[i].to==fa[x][0])continue;
        deep[edge[i].to]=deep[x]+1;
        fa[edge[i].to][0]=x;
        dfs(edge[i].to);
    }
}
int gets(int a,int b)
{
    for(int i=0;i<=20;i++)
        if(b&(1<<i))a=fa[a][i];
    return a;
}
int lca(int u,int v)
{
    if(deep[u]<deep[v])swap(u,v);
    u=gets(u,deep[u]-deep[v]);
    if(u==v)return u;
    for(int i=15;i>=0;i--)
        if(fa[u][i]!=fa[v][i])
            u=fa[u][i],v=fa[v][i];
    return fa[u][0];
}
int main()
{
    int x,y;
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
    {
        scanf("%d%d",&x,&y);
        if(y==-1)root=x;
        else add(x,y),add(y,x);
    }
    scanf("%d",&m);
    dfs(root);
    for(int i=1;i<=m;i++)
    {
        scanf("%d%d",&x,&y);
        if(x==y)printf("0\n");
        else
        {
            int l=lca(x,y);
            if(l==x) printf("1\n");
            else
                if(l==y) printf("2\n");
                else printf("0\n");
        }
    }
}