相交的铁路线

题目名称：相交的铁路线

题目链接：相交的铁路线

描述

H国有 n 个城市，编号1…n。城市间有n-1条铁路，保证任意两个城市可以通过铁路互达，且路线唯一。

现有 m 次询问，每次询问两条铁路线是否相交（有共同经过的城市或铁路）。

输入

第一行一个数 T，表示数据组数

对于每一组数据：

第一行两个数n, m

第2~n行，每行两个数x, y表示有一条铁路连接城市 x 和 y

接下来m行每行四个数，x1, y1, x2, y2 表示询问城市 x1 和 y1 之间的路线是否和城市 x2 和 y2 之间的路线相交。

对于40%的数据，1 ≤ n, m ≤ 100

对于60%的数据，1 ≤ n, m ≤ 1000

对于100%的数据，1 ≤ T ≤ 10, 1 ≤ n, m ≤ 100000 1 ≤ x, y, x1, y1, x2, y2 ≤ n

输出

对于每次询问输出YES或NO

样例输入

1
4 2
1 2
2 3
3 4
1 2 3 4
1 4 2 3

样例输出

NO
YES

解题思路

难度不大，最近公共祖先问题，注意因为数据范围比较大，所以需要提前进行预处理，采用《挑战程序设计竞赛》中的LCA模板，再结合讨论中需要注意判断的地方就可以了

完整代码

#include<bits/stdc++.h>
using namespace std;
#define MAX_V 100010
#define MAX_LOG_V 20   //取对数
int T,n,m,x,y,a,b;
vector<int> g[MAX_V];
int f[MAX_LOG_V][MAX_V],depth[MAX_V];

void init(){
	for(int i=1;i<=n;i++){
		g[i].clear();
	}
}

void dfs(int v,int p,int d){
	f[0][v]=p;
	depth[v]=d;
	for(int i=0;i<g[v].size();i++){
		if(g[v][i]!=p){
			dfs(g[v][i],v,d+1);
		}
	}
}

int lca(int u,int v){
	if(depth[u]>depth[v]) swap(u,v);
	for(int k=0;k<MAX_LOG_V;k++){
		if((depth[v]-depth[u])>>k&1){
			v=f[k][v];
		}
	}
	if(u==v) return u;
	for(int k=MAX_LOG_V-1;k>=0;k--){
		if(f[k][u]!=f[k][v]){
			u=f[k][u];
			v=f[k][v];
		}
	}
	return f[0][u];
}

int main()
{
	cin>>T;
	while(T--){
		cin>>n>>m;
		init();
		for(int i=1;i<n;i++){
			cin>>x>>y;
			if(x>y) swap(x,y);
			g[x].push_back(y);
		}
		
		dfs(1,-1,0);
		for(int k=0;k+1<MAX_LOG_V;k++){
			for(int v=0;v<=n;v++){
				if(f[k][v]<0) f[k+1][v]=-1;
				else f[k+1][v]=f[k][f[k][v]];
			}
		}
		
		for(int i=0;i<m;i++){
			cin>>x>>y>>a>>b;
			int t1=lca(x,y);
			int t2=lca(a,b);
			if(lca(t1,t2)==t1){
				if(lca(t2,x)==t2||lca(t2,y)==t2) puts("YES");
					 else puts("NO");
			}
		 	else if(lca(t1,a)==t1||lca(t1,b)==t1) puts("YES");
		    else puts("NO");
		}
	}
	return 0;
}

未进行预处理的程序如下（当然超时了）

#include<bits/stdc++.h>
using namespace std;
#define MAX_V 100010
int T,n,m,x,y,a,b;
vector<int> g[MAX_V];
int f[MAX_V],depth[MAX_V];

void init(){
	for(int i=1;i<=n;i++){
		g[i].clear();
		f[i]=depth[i]=0;
	}
}

void dfs(int v,int p,int d){
	f[v]=p;
	depth[v]=d;
	for(int i=0;i<g[v].size();i++){
		if(g[v][i]!=p){
			dfs(g[v][i],v,d+1);
		}
	}
}

int lca(int u,int v){
	while(depth[u]>depth[v]) u=f[u];
	while(depth[v]>depth[u]) v=f[v];
	
	while(u!=v){
		u=f[u];
		v=f[v];
	}
	return u;
}

int main()
{
	cin>>T;
	while(T--){
		cin>>n>>m;
		init();
		for(int i=1;i<n;i++){
			cin>>x>>y;
			if(x>y) swap(x,y);
			g[x].push_back(y);
		}
		dfs(1,-1,0);
		for(int i=0;i<m;i++){
			cin>>x>>y>>a>>b;
			int t1=lca(x,y);
			int t2=lca(a,b);
			if(lca(t1,t2)==t1){
				if(lca(t2,x)==t2||lca(t2,y)==t2) puts("YES");
					 else puts("NO");
			}
		 	else if(lca(t1,a)==t1||lca(t1,b)==t1) puts("YES");
		    else puts("NO");
		}
	}
	return 0;
}