题目
输入输出
样例
输入1:
8
1 2
1 3
2 4
2 5
5 6
5 7
3 8
4
2 5 4 3
5 3 8 8
5 4 6 7
4 8 6 7
输入2:
15
2 1
3 1
4 2
5 3
6 2
7 2
8 5
9 3
10 6
11 5
12 7
13 11
14 1
15 1
5
1 2 3 4
4 7 1 9
2 3 7 9
2 6 7 8
2 1 6 8
输出1:
YES
NO
YES
NO
输出2:
YES
NO
YES
YES
YES
数据范围
分析
这道题实际就是求树上两条链是否相交,对此,我们通过手玩样例可得,当一条链的lca在另一条链上时相交,这样就是一道套路题了。
上代码
#include<bits/stdc++.h>
using namespace std;
int ver[200010],nxt[200010],hed[200010],tot,f[100010][20],d[100010],t,n,qq;
queue<int> q;
void add(int x,int y){
ver[++tot]=y,nxt[tot]=hed[x],hed[x]=tot;
}
void bfs(){
q.push(1);
d[1]=1;
while(q.size()){
int x=q.front();
q.pop();
for(int i=hed[x];i;i=nxt[i]){
int y=ver[i];
if(d[y])
continue;
d[y]=d[x]+1;
f[y][0]=x;
for(int j=1;j<=t;j++)
f[y][j]=f[f[y][j-1]][j-1];
q.push(y);
}
}
}
int lca(int x,int y){
if(d[x]>d[y])
swap(x,y);
for(int i=t;i>=0;i--){
if(d[f[y][i]]>=d[x])
y=f[y][i];
}
if(x==y)
return x;
for(int i=t;i>=0;i--)
if(f[x][i]!=f[y][i])
x=f[x][i],y=f[y][i];
return f[x][0];
}
bool pd(int x,int y,int z,int fa){
if((d[x]>max(d[y],d[z]))||(d[x]<d[fa]))
return 0;
if(d[x]>=d[y]){
if(x==y)
return 1;
for(int i=t;i>=0;i--)
if(d[x]<=d[f[z][i]])
z=f[z][i];
if(z==x)
return 1;
return 0;
}
if(d[x]>=d[z]){
if(x==z)
return 1;
for(int i=t;i>=0;i--)
if(d[x]<=d[f[y][i]])
y=f[y][i];
if(x==y)
return 1;
return 0;
}
for(int i=t;i>=0;i--)
if(d[x]<=d[f[y][i]])
y=f[y][i];
for(int i=t;i>=0;i--)
if(d[x]<=d[f[z][i]])
z=f[z][i];
if((x==y)||(x==z))
return 1;
return 0;
}
int main(){
scanf("%d",&n);
t=(int)(log(n)/log(2))+1;
for(int i=1,x,y;i<n;i++){
scanf("%d%d",&x,&y);
add(x,y),add(y,x);
}
bfs();
scanf("%d",&qq);
while(qq--){
int a,b,c,d;
scanf("%d%d%d%d",&a,&b,&c,&d);
int qaq=lca(a,b),qwq=lca(c,d);
if(pd(qaq,c,d,qwq)||pd(qwq,a,b,qaq))
puts("YES");
else
puts("NO");
}
return 0;
}