P3398 仓鼠找sugar

题目

题目

思路

经过手玩样例发现，2条路径相交，当且仅当其中一条的LCA在另一条路径上
然后根据结论写代码
code:

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<string>
#include<cstring>
#include<cmath> 
using namespace std;
int fa[500010][27],head[500010],c[500010],x,y,xx,yy;
struct f{
	int to,net;
} a[1000020];
int n,m,s;
void dfs(int x,int f)
{
	fa[x][0]=f;
	c[x]=c[f]+1;
	for (int j=1;(1<<j)<=c[x];j++)
	{
		fa[x][j]=fa[fa[x][j-1]][j-1];
	}
	for (int i=head[x];i!=-1;i=a[i].net)
	{
		if (a[i].to!=f) dfs(a[i].to,x);
	}
	return;
}
int k;
void add(int u,int v)
{
    a[k].to=v;
	a[k].net=head[u];
    head[u]=k++;
    return; 
}
int LCA(int x,int y)
{
	if (c[x]>c[y]) swap(x,y);
	for (int i=20;i>=0;i--)
	{
		if (c[x]<=c[y]-(1<<i)) y=fa[y][i];
	}
	if (x==y) return x;
	for (int i=20;i>=0;i--)
	{
		if (fa[x][i]==fa[y][i]) continue;
		else x=fa[x][i],y=fa[y][i];
	}
	return fa[x][0];
}
int main()
{
    memset(head,-1,sizeof(head));
	cin>>n>>m;
	for (int i=1;i<n;i++)
	{
		cin>>x>>y;
		add(x,y);
		add(y,x);
	}
	dfs(1,0);
	while (m--)
	{
		cin>>x>>xx>>y>>yy;
		int z=LCA(x,xx),zz=LCA(y,yy);
		if (c[z]<c[zz])
		{
			swap(z,zz);
			swap(x,y);
			swap(xx,yy);
		}
		if (LCA(z,y)==z||LCA(z,yy)==z) cout<<"Y"<<endl;
		else cout<<'N'<<endl;
	}
    return 0;
}