模拟赛 祖孙询问(时间限制:1s;空间限制:128MB)

【问题描述】

    已知一棵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。

【样例输入】

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

0

0

0

2

【数据规模】

    对于30%的数据,n,m≤1000。

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


题解

我写了LCA,BFS序更快。

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<iostream>
#include<algorithm>
using namespace std;
int n,m,zz,head[40002],root;
struct bian{int to,nx;} e[80002];
int fa[40002][20],h[40002];//2^16=65536
void insert(int x,int y)
{
	zz++; e[zz].to=y; e[zz].nx=head[x]; head[x]=zz;
	zz++; e[zz].to=x; e[zz].nx=head[y]; head[y]=zz;
}
void init()
{
	int i,x,y;
	scanf("%d",&n);
	for(i=1;i<=n;i++)
	   {scanf("%d%d",&x,&y);
	    if(y==-1) root=x;
	    else insert(x,y);
	   }
}
void dfs(int x)
{
	int i,p;
	for(i=1;i<=15;i++)
	   {if(h[x]<(1<<i)) break;
	    fa[x][i]=fa[fa[x][i-1]][i-1];
	   }
	for(i=head[x];i;i=e[i].nx)
	   {p=e[i].to;
	    if(p==fa[x][0]) continue;
	    h[p]=h[x]+1; fa[p][0]=x;
	    dfs(p);
	   }
}
int lca(int x,int y)
{
	if(h[x]<h[y]) swap(x,y);
	int i,c;
	c=h[x]-h[y];
	for(i=15;i>=0;i--)
	   {if(c&(1<<i)) x=fa[x][i];}
	for(i=15;i>=0;i--)
	   {if(fa[x][i]!=fa[y][i])
	       {x=fa[x][i]; y=fa[y][i];}
	   }
	if(x==y) return x;
	else return fa[x][0];
}
void work()
{
	int i,x,y,t;
	scanf("%d",&m);
	dfs(root);
	for(i=1;i<=m;i++)
	   {scanf("%d%d",&x,&y);
	    if(x==y) printf("0\n");
		else
		   {t=lca(x,y);
		    if(t==x) printf("1\n");
		    else if(t==y) printf("2\n");
		    else printf("0\n");
		   }
	   }
}
int main()
{
	freopen("tree.in","r",stdin);
	freopen("tree.out","w",stdout);
	init(); work();
	return 0;
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值