样例过了,爆0,求助大佬
查看原帖
样例过了,爆0,求助大佬
570507
witw楼主2022/2/4 16:28
#include<iostream>
#include<cstring>
using namespace std;
const int N=1e5+5;
int h[2*N],ne[2*N],e[2*N],idx;
int f[N][21],dep[N];
int n,m;
void add(int x,int y){
	e[idx]=x,ne[idx]=h[y],h[y]=idx++;
}
void dfs(int u,int v){
	f[u][0]=v,dep[u]=dep[v]+1;
	for(int i=1;(1<<i)<=dep[u];++i){
		f[u][i]=f[f[u][i-1]][i-1];
	}
	for(int i=h[u];i!=-1;i=ne[i]){
		int k=e[i];
		if(k==v)return;
		dfs(k,u);
	}
}
inline int lca(int x,int y){
	if(dep[x]<dep[y])swap(x,y);
	for(int i=20;i>=0;i--){
		if(dep[f[x][i]]>=dep[y])x=f[x][i];
		if(x==y)return x;
	}
	for(int i=20;i>=0;i--){
		if(f[x][i]!=f[y][i]){
			x=f[x][i],y=f[y][i];
		}
	}
	return f[x][0];
}
inline int abs(int x) { return x > 0 ? x : -x; }
inline int dis(int a,int b){
	int c=lca(a,b);
	return abs(dep[c]-dep[a])+abs(dep[c]-dep[b]);
}
int main(){
	memset(h,-1,sizeof h);
	cin>>n>>m;
	for(int i=1;i<n;++i){
		int x,y;
		cin>>x>>y;
		add(x,y);
		add(y,x);
	} 
	dfs(1,0);
	for(int i=1;i<=m;++i){
		int a,b,c,d;
		cin>>a>>b;
		cin>>c>>d;
		int x=lca(a,b),y=lca(c,d);
		if(dis(a,y)+dis(b,y)==dis(a,b)||dis(c,x)+dis(d,x)==dis(c,d))cout<<"Y"<<endl;
		else cout<<"N"<<endl;
	}
	return 0;
}
2022/2/4 16:28
加载中...