T70138 寻找中点
题目描述已知一棵n个节点的有根树。有m个询问。每个询问给出了一对节点的编号x 和 y,询问x与 y的祖孙关系。 输入输出格式 输入格式:
输入第一行包括一个整数n表示节点个数。
接下来n行每行一对整数对a和b 表示a和 b之间有连边。如果 b是-1,那么 a就是树的根。
第 n+2 行是一个整数 m表示询问个数。
接下来m行,每行两个正整数x和 y。
输出格式:
对于每一个询问,输出1如果x是y的祖先,输出2如果y是x的祖先,否则输出0。
输入输出样例 输入样例#1: 复制
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输出样例#1: 复制
1 0 0 0 2
说明
【样例说明】
234是233的祖先,19是233的祖先。
【数据范围】
对30%的输入数据 :n,m≤1000
对100%的输入数据 :n,m≤40000,每个节点的编号大于0,都不超过 40000。
#include<bits/stdc++.h>
using namespace std;
const int N=100009;
int n,q,x,y,geng,first[N*2],next[N*2],to[N*2],dep[N],f[N][21],tot;
void add(int u,int v)
{
next[++tot]=first[u];first[u]=tot;to[tot]=v;
next[++tot]=first[v];first[v]=tot;to[tot]=u;
}
void Init(int u,int father)
{
dep[u]=dep[father]+1;
for(int i=0;i<=19;i++)
{
f[u][i+1]=f[f[u][i]][i];
}
for(int e=first[u];e;e=next[e])
{
int v=to[e];
if(v==father) continue;
f[v][0]=u;
Init(v,u);
}
}
int find(int x,int y)
{
if(dep[x]<dep[y]) swap(x,y);
for(int i=20;i>=0;i--)
{
if(dep[f[x][i]]>=dep[y]) x=f[x][i];
if(x==y) return x;
}
for(int i=20;i>=0;i--)
{
if(f[x][i]!=f[y][i])
{
x=f[x][i];
y=f[y][i];
}
}
return f[x][0];
}
int main()
{
cin>>n;
for(int i=1;i<=n;i++)
{
cin>>x>>y;
if(y==-1)
{
geng=x;
continue;
}
add(x,y);
}
Init(geng,0);
cin>>q;
for(int i=1;i<=q;i++)
{
cin>>x>>y;
int ans=find(x,y);
if(ans==x) cout<<1<<endl;
else if(ans==y) cout<<2<<endl;
else cout<<0<<endl;
}
return 0;
}