jzoj3054-祖孙询问【LCA】

正题

---

题目大意

求两个点的祖孙关系。

---

解题思路

直接$LCA$，然后暴力判断。

---

$code$

#include<cstdio>
#include<algorithm>
#include<queue>
#include<cmath>
#define N 41000
using namespace std;
struct line{
    int to,next,w;
}a[N*5];
int tot,n,m,s,x,y,ls[N],dep[N],f[N][30],dis[N][30],t;
queue<int> q;
inline int readn()
{
    int X=0,w=0; char c=0;
    while(c<'0'||c>'9') {w|=c=='-';c=getchar();}
    while(c>='0'&&c<='9') X=(X<<3)+(X<<1)+(c^48),c=getchar();
    return w?-X:X;
}
inline void addl(int x,int y,int w)
{
    a[++tot].to=y;
    a[tot].next=ls[x];
    a[tot].w=w;
    ls[x]=tot;
}
inline void bfs(int s)
{
    q.push(s);dep[s]=1;
    while(!q.empty())
    {
        int x=q.front();q.pop();
        for (int i=ls[x];i;i=a[i].next)
        {
            int y=a[i].to;
            if (dep[y]) continue;
            q.push(y);f[y][0]=x;
            dep[y]=dep[x]+1;
            dis[y][0]=a[i].w;
        }
    }
    t=(int)(log(n)/log(2))+1;
    for (int j=1;j<=t;j++)
    {
      for (int i=1;i<=40000;i++)
      {
        f[i][j]=f[f[i][j-1]][j-1];
        dis[i][j]=min(dis[i][j-1],dis[f[i][j-1]][j-1]);
      }
    }
}
inline int LCA(int x,int y)
{
    if (dep[x]>dep[y]) swap(x,y);
    for (int i=t;i>=0;i--)
      if (dep[f[y][i]]>=dep[x]) y=f[y][i];
    if (x==y) return x;
    for (int i=t;i>=0;i--)
      if (f[y][i]!=f[x][i]) 
      {
        x=f[x][i];
        y=f[y][i];
      }
    return f[x][0];
}
int main()
{
    n=readn();
    for(int i=1;i<=n;i++)
    {
    	x=readn();y=readn();
    	if(y==-1) s=x;
    	else
    	{
        	addl(x,y,1);
        	addl(y,x,1);
    	}
    }
    m=readn();
    bfs(s);
    for(int i=1;i<=m;i++)
    {
    	x=readn();y=readn();
        int lca=LCA(x,y);
        if(lca==x) printf("1\n");
        else if(lca==y) printf("2\n");
        else printf("0\n");
    }
}