LCA 模板 JZOJ 3054. 【NOIP2012模拟10.27】祖孙询问

//luo
 1 #include<iostream>
#include<vector>
#include<cstdio>
using namespace std;
int n,root,m;
vector <int> f[1000100];
int depth[1000100];
int fa[10000100];
inline int read()
{
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
void dfs(int fath,int now)
{
    depth[now]=depth[fath]+1;
    fa[now]=fath;
    for (int i=0;i<f[now].size();i++)
        dfs(now,f[now][i]);
}
int LCA(int x,int y)
{
    if (depth[y]>depth[x])
        swap(x,y);
    while (depth[x]>depth[y])
       x=fa[x];
    while (x!=y)
      x=fa[x],y=fa[y];
    return x;
}
int main ()
{
    
    cin>>n;
    for (int i=1,x,y;i<=n;i++)
    {
        x=read(); y=read();
        if (y==-1)
          root=x;
        f[y].push_back(x);
    }
    depth[0]=-1;
    dfs(0,root);
    cin>>m;
    for (int i=1,x,y;i<=m;i++)
    {
        x=read(); y=read();
        int s=LCA(x,y);
        if (s==x) cout<<1<<endl;
        else if (s==y) cout<<2<<endl;
        else
           cout<<0<<endl;
    }
}

 倍增优化

 我们每次可以不仅仅局限于一步一步的跳

 原来加上记录他跳几次的点就好了

#include<iostream>
#include<vector>
#include<cstdio>
using namespace std;
int n,root,m;
vector <int> f[100010];
int depth[100010];
int fa[1000010][22];
inline int read()
{
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
void dfs(int fath,int now)
{
    depth[now]=depth[fath]+1;
    fa[now][0]=fath;
    for (int i=1;i<21;i++)
       fa[now][i]=fa[fa[now][i-1]][i-1];
    for (int i=0;i<f[now].size();i++)
        dfs(now,f[now][i]);
}
int LCA(int x,int y)
{
    if (depth[y]>depth[x])
        swap(x,y);
    for (int i=20;i>=0;i--)
       if (depth[x]-(1<<i)>=depth[y])
         x=fa[x][i];
    if (x==y) return x;
    for (int i=20;i>=0;i--)
       if (fa[x][i]!=fa[y][i])
         x=fa[x][i],y=fa[y][i];
    return fa[x][0];
}
int main ()
{
    
    cin>>n;
    for (int i=1,x,y;i<=n;i++)
    {
        x=read(); y=read();
        if (y==-1)
          root=x;
        f[y].push_back(x);
    }
    depth[0]=-1;
    dfs(0,root);
    cin>>m;
    for (int i=1,x,y;i<=m;i++)
    {
        x=read(); y=read();
        int s=LCA(x,y);
        if (s==x) cout<<1<<endl;
        else if (s==y) cout<<2<<endl;
        else
           cout<<0<<endl;
    }
}

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。

 

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

 

Data Constraint

 

 

Hint

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