P3398 仓鼠找 sugar

//https://www.luogu.com.cn/problem/P3398
#include <bits/stdc++.h>
using namespace std;
const int N=1e5+10,M=N*2;
int e[M],ne[M],h[N],idx;
int depth[N],fa[N][20];
int q[N];
int n,m;
int root;
void add(int a,int b){
    e[idx]=b,ne[idx]=h[a],h[a]=idx++;
}
void bfs(int root){
    memset(depth,0x3f,sizeof depth);
    depth[0]=0,depth[root]=1;
    int hh=0,tt=1;
    q[0]=root;
    while(hh<tt){
        int u=q[hh++];
        for(int i=h[u];~i;i=ne[i]){
            int v=e[i];
            if(depth[v]>depth[u]+1){
                depth[v]=depth[u]+1;
                q[tt++]=v;
                fa[v][0]=u;
                for(int k=1;k<=19;k++) fa[v][k]=fa[fa[v][k-1]][k-1];
            }
        }
    }
}
int lca(int a,int b){
    if(depth[a]<depth[b]) swap(a,b);
    for(int k=19;k>=0;k--) if(depth[fa[a][k]]>=depth[b]) a=fa[a][k];
    if(a==b) return a;
    for(int k=19;k>=0;k--){
        if(fa[a][k]!=fa[b][k]){
            a=fa[a][k],b=fa[b][k];
        }
    }
    return fa[a][0];
}
bool isOnPath(int u,int v,int p){
	int anc=lca(u,v);
	return (lca(u,p)==p&&lca(v,p)==anc)||(lca(v,p)==p&&lca(u,p)==anc);
}
int main(){
    scanf("%d%d",&n,&m);
    memset(h,-1,sizeof h);
    for(int i=0,a,b;i<n-1;i++){
        scanf("%d%d",&a,&b);
        add(a,b),add(b,a);
    }
    int root=1;
    bfs(root);
    for(int i=0,a,b,c,d;i<m;i++){
        scanf("%d%d%d%d",&a,&b,&c,&d);
        int anc1=lca(a,b),anc2=lca(c,d);
        bool cond=isOnPath(c,d,a)||isOnPath(c,d,b)||isOnPath(a,b,c)||isOnPath(a,b,d)||isOnPath(c,d,anc1)||isOnPath(a,b,anc2);
        puts(cond?"Y":"N");
    }
    return 0;
}
posted @ 2025-05-05 12:12  九三青梧  阅读(10)  评论(0)    收藏  举报