【NOIP2018提高组模拟9.6】相交 题解

题目

输入输出

样例
输入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;
}