洛谷P3398 仓鼠找sugar

题目描述

小仓鼠的和他的基（mei）友（zi）sugar住在地下洞穴中，每个节点的编号为1~n。地下洞穴是一个树形结构。这一天小仓鼠打算从从他的卧室（a）到餐厅（b），而他的基友同时要从他的卧室（c）到图书馆（d）。他们都会走最短路径。现在小仓鼠希望知道，有没有可能在某个地方，可以碰到他的基友？

小仓鼠那么弱，还要天天被zzq大爷虐，请你快来救救他吧！

输入输出格式

输入格式：

第一行两个正整数n和q，表示这棵树节点的个数和询问的个数。

接下来n-1行，每行两个正整数u和v，表示节点u到节点v之间有一条边。

接下来q行，每行四个正整数a、b、c和d，表示节点编号，也就是一次询问，其意义如上。

输出格式：

对于每个询问，如果有公共点，输出大写字母“Y”；否则输出“N”。

输入输出样例

输入样例#1： 

5 5
2 5
4 2
1 3
1 4
5 1 5 1
2 2 1 4
4 1 3 4
3 1 1 5
3 5 1 4

输出样例#1： 

Y
N
Y
Y
Y

说明

__本题时限1s，内存限制128M。

20%的数据 n<=200,q<=200

40%的数据 n<=2000,q<=2000

70%的数据 n<=50000,q<=50000

100%的数据 n<=100000,q<=100000

典型的 LCA 问题，跑倍增就好了。

附代码：

#include<iostream>
#include<algorithm>
#include<cstdio>
#define MAXN 100010
using namespace std;
int n,m,c=1;
int head[MAXN<<1],deep[MAXN],f[MAXN][20];
struct node{
	int next,to;
}a[MAXN<<1];
inline int read(){
	int date=0,w=1;char c=0;
	while(c<'0'||c>'9'){if(c=='-')w=-1;c=getchar();}
	while(c>='0'&&c<='9'){date=date*10+c-'0';c=getchar();}
	return date*w;
}
void add(int x,int y){
	a[c].to=y;
	a[c].next=head[x];
	head[x]=c++;
	a[c].to=x;
	a[c].next=head[y];
	head[y]=c++;
}
void buildtree(int rt){
	int will;
	for(int i=head[rt];i;i=a[i].next){
		will=a[i].to;
		if(!deep[will]){
			deep[will]=deep[rt]+1;
			f[will][0]=rt;
			buildtree(will);
		}
	}
}
void step(){
	for(int i=1;i<=19;i++)
	for(int j=1;j<=n;j++)
	f[j][i]=f[f[j][i-1]][i-1];
}
int LCA(int x,int y){
	if(deep[x]<deep[y])swap(x,y);
	for(int i=19;i>=0;i--)
	if(deep[f[x][i]]>=deep[y])
	x=f[x][i];
	if(x==y)return x;
	for(int i=19;i>=0;i--)
	if(f[x][i]!=f[y][i]){
		x=f[x][i];
		y=f[y][i];
	}
	return f[x][0];
}
int main(){
	int a,b,c,d;
	n=read();m=read();
	for(int i=1;i<n;i++){
		a=read();b=read();
		add(a,b);
	}
	deep[1]=1;
	buildtree(1);
	step();
	for(int i=1;i<=m;i++){
		a=read();b=read();c=read();d=read();
		int fa1=LCA(a,b),fa2=LCA(c,d);
		if(deep[fa1]<deep[fa2]){
			swap(a,c);swap(b,d);
			swap(fa1,fa2);
		}
		if(LCA(fa1,c)==fa1||LCA(fa1,d)==fa1)printf("Y\n");
		else printf("N\n");
	}
	return 0;
}