祖孙询问_纪中3054_lca

Description

---

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

Input

---

输入第一行包括一个整数n表示节点个数。

接下来n行每行一对整数对a和b表示a和b之间有连边。如果b是-1，那么a就是树的根。

第n+2行是一个整数m表示询问个数。

接下来m行，每行两个正整数x和y。

Output

---

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

Hint

---

对于100%的.据,n,m≤40000，每个节点的编号都不超过40000。

Analysis

---

赤裸裸的LCA，数组开小我的锅，我的锅

Code

---

#include <stdio.h>
#include <cstring>
using namespace std;
struct edge
{
    int x,y,next;
};
bool v[80501];
edge e[80501],g[80501];
int ans[80501],lse[80501],lsg[80501],f[80501];
int maxG,maxE;
int find(int x)
{
    if (f[x]==-1)
        return x;
    f[x]=find(f[x]);
    return f[x];
}
void dfs(int x)
{
    v[x]=true;
    for (int i=lsg[x];i;i=g[i].next)
        if (v[g[i].y])
            ans[i]=find(g[i].y);
    for (int i=lse[x];i;i=e[i].next)
        {
            dfs(e[i].y);
            f[find(e[i].y)]=x;
        }
}
int main()
{
    memset(f,-1,sizeof(f));
    int n,m,root;
    scanf("%d",&n);
    for (int i=1;i<=n;i++)
    {
        int x,y;
        scanf("%d%d",&x,&y);
        if (y==-1)
            root=x;
        else
        {
            e[++maxE]=(edge){y,x,lse[y]};
            lse[y]=maxE;
        }
    }
    scanf("%d",&m);
    for (int i=1;i<=m;i++)
    {
        int x,y;
        scanf("%d%d",&x,&y);
        g[++maxG]=(edge){x,y,lsg[x]};
        lsg[x]=maxG;
        g[++maxG]=(edge){y,x,lsg[y]};
        lsg[y]=maxG;
    }
    dfs(root);
    for (int i=1;i<=m;i++)
    {
        edge now=g[i*2-1];
        if (ans[i*2-1]==now.x||ans[i*2]==now.x)
            printf("1\n");
        else
            if (ans[i*2-1]==now.y||ans[i*2]==now.y)
                printf("2\n");
                else
                    printf("0\n");
    }
    return 0;
}