BNUOJ 52509 Borrow Classroom

最近公共祖先。

如果$A$到$1$的时间小于$B$到$C$再到$1$的时间，那么一定可以拦截。

如果上述时间相等，需要在到达$1$之前，两者相遇才可以拦截。

#include<bits/stdc++.h>
using namespace std;
int T,n,Q,sz;
int dep[],f[];
int dp[][];
int h[],to[],nx[];
void add(int x,int y)
{
    to[sz] = y;
    nx[sz] = h[x];
    h[x] = sz++;
}
int LCA(int x,int y)
{
    if(dep[x]<dep[y]) swap(x,y);
    while()
    {
        if(dep[x]==dep[y]) break;
        for(int j=;j>=;j--)
        {
            int p = dp[x][j];
            if(p==-) continue;
            if(dep[p]<dep[y]) continue;
            x=p;
            break;
        }
    }
    if(x==y) return x;
    while()
    {
        if(dp[x][]==dp[y][]) return dp[x][];
        for(int j=;j>=;j--)
        {
            int p = dp[x][j];
            int q = dp[y][j];
            if(p==q) continue;
            x = p;
            y = q;
            break;
        }
    }
}
int dis(int x,int y)
{
    int t = LCA(x,y);
    return dep[x]-dep[t] + dep[y]-dep[t];
}
void dfs(int x,int y,int d)
{
    f[x]=; dp[x][] = y; dep[x] = d;
    for(int i = h[x];i!=-;i=nx[i])
    {
        int p = to[i];
        if(f[p]) continue;
        dfs(p,x,d+);
    }
}
int main()
{
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d%d",&n,&Q);
        for(int i=;i<=n;i++)
        {
            f[i]=;
            h[i]=-;
        }
        sz=;
        for(int i=;i<=n-;i++)
        {
            int x,y; scanf("%d%d",&x,&y);
            add(x,y); add(y,x);
        }
        dfs(,-,);
        for(int j=;j<=;j++)
            for(int i=;i<=n;i++)
            {
                if(dp[i][j-]==-) dp[i][j]=-;
                else dp[i][j] = dp[dp[i][j-]][j-];
            }
        for(int i=;i<=Q;i++)
        {
            int A,B,C; scanf("%d%d%d",&A,&B,&C);
            int ans=;
            int disAC = dis(A,C);
            int disBC = dis(B,C);
            int disA1 = dep[A]-dep[];
            int disC1 = dep[C]-dep[];
            if(disAC<=disBC) ans=;
            if(disA1<disBC+disC1) ans=;
            if(disA1==disBC+disC1&&LCA(A,C)!=) ans=;
            if(ans==) printf("YES\n");
            else printf("NO\n");
        }
    }
    return ;
}