70pts求条
查看原帖
70pts求条
1306142
Eddy1341楼主2024/9/28 10:58

记录详情看这里

#include<iostream>
#include<vector>
#include<cmath>
using namespace std;
const int N=1e5+5,D=log2(N);
int d;
vector<int> g[N];
int f[N],dp[N][D],dep[N];
void dfs(int x,int fa){
	dep[x]=dep[fa]+1;
	dp[x][0]=fa;
	for(int i=1;i<=d;i++){
		dp[x][i]=dp[dp[x][i-1]][i-1];
	}
	for(auto y:g[x]){
		if(y!=fa)dfs(y,x);
	}
}
int lca(int x,int y){
	if(dep[x]<dep[y])swap(x,y);
	for(int i=d;i>=0;i--)
		if(dep[dp[x][i]]>=dep[y])
			x=dp[x][i];
	if(x==y)return x;
	for(int i=d;i>=0;i--)
		if(dp[x][i]!=dp[y][i]){
			x=dp[x][i];
			y=dp[y][i];
		}
	return dp[x][0];
}
int dis(int x,int y){
	int z=lca(x,y);
	return abs(dep[x]-dep[z])+abs(dep[y]-dep[z]);
}
int main(){
	int n,q;
	scanf("%d%d",&n,&q);
	d=log2(n);
	while(n-->1){
		int u,v;
		scanf("%d%d",&u,&v);
		g[u].push_back(v);
		g[v].push_back(u);
	}
	dfs(1,0);
	while(q--){
		int a,b,c,d;
		scanf("%d%d%d%d",&a,&b,&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))puts("Y");
		else puts("N");
	}
	return 0;
}
2024/9/28 10:58
加载中...