祖孫詢問
Time Limit: 1 Sec Memory Limit: 128 MBSubmit: 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
接下來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");
}
}
}