LCA全WA求调
查看原帖
LCA全WA求调
809371
Love_Natsu楼主2024/8/13 10:36

阳历过了

#include<iostream>
#include<algorithm>
#define N 100005
using namespace std;
int read(){
	char c=getchar();
	int x=0,f=1;
	for(;c<'0'||c>'9';c=getchar())if(c=='-')f*=-1;
	for(;c>='0'&&c<='9';c=getchar())x=x*10+c-'0';
	return f*x;
}
int n,q;
struct edge{int v,prv;}e[N<<1];
int otto,h[N];
void add(int u,int v){
	e[++otto]={v,h[u]};
	h[u]=otto;
}
int fa[N][20],dep[N];
void dfs(int x,int f){
	fa[x][0]=f,dep[x]=dep[f]+1;
	for(int i=1;i<=17;i++){
		fa[x][i]=fa[fa[x][i-1]][i-1];
	}
	for(int i=h[x];i;i=e[i].prv){
		if(e[i].v==f)continue;
		dfs(e[i].v,x);
	}
}
int lca(int x,int y){
	if(x>y)swap(x,y);
	for(int i=17;i>=0;i--)
		if(dep[fa[x][i]]>=dep[y])x=fa[x][i];
	if(x==y)return x;
	for(int i=17;i>=0;i--)
	    if(fa[x][i]!=fa[y][i])x=fa[x][i],y=fa[y][i];
	return fa[x][0];
}
int dis(int x,int y){
	int lxy=lca(x,y);
	return abs(dep[x]-dep[lxy])+abs(dep[y]-dep[lxy]);;
}
int main(){
	n=read(),q=read();
	for(int i=1;i<n;i++){
		int u=read(),v=read();
		add(u,v),add(v,u);
	}
	dfs(1,0);
	while(q--){
		int a=read(),b=read(),c=read(),d=read();
		int lab=lca(a,b),lcd=lca(c,d);
	    if(dis(a,b)==dis(lcd,a)+dis(lcd,b)||dis(c,d)==dis(lab,c)+dis(lab,d))putchar('Y');
	    else putchar('N');
		puts("");
	}
	return;
}
2024/8/13 10:36
加载中...