CodeForces - 1304E 1-Trees and Queries（LCA求树上距离）

🥚 🥚 🥚

问你x,y点连接之后a,b是否可以经过k长的路到达彼此，路径可以重复走。

a要走到b只有三种走法：
1，直接从a到b
2，a到x，x到y，y到b
3，a到y，y到x，x到b

对于每种走法，检查他是否大于等于k且奇偶性相同，因为我们可以通过在两点之间“反复横跳”达到k（比如k=5，len=3，选路径上两个点来回走一次即可）

// lca
int dep[MAXN],dis[MAXN],f[MAXN][22];
//链式向前星
int n,tot,head[MAXN];
struct IN{int v,w,next;}edge[MAXN<<1];
void addedge(int u,int v,int w)
{
    edge[tot].v=v,edge[tot].w=w,edge[tot].next=head[u];
    head[u]=tot++;
}
void build()
{
    tot=1;
    memset(head,0,sizeof(head));
    for(int i=1; i<n; i++)
    {
        int u,v,w=1; scanf("%d %d",&u,&v);
        addedge(u,v,w);addedge(v,u,w);
    }
}
void lca_dfs(int u,int pre)
{
    for(int i=1; i<=20; i++)
    {
        f[u][i]=f[f[u][i-1]][i-1];
        if(!f[u][i])   break;
    }
    for(int i=head[u]; i; i=edge[i].next)
    {
        int v=edge[i].v;
        if(v==pre)  continue;
        dep[v]=dep[u]+1,dis[v]=dis[u]+edge[i].w, f[v][0]=u;
        lca_dfs(v,u);
    }
}
void lca_init()
{
    dep[0]=-1,dep[1]=1,dis[1]=0;
    memset(dis,0,sizeof(dis));
    lca_dfs(1,-1);
}
int lca(int x,int y)
{
    if(dep[x]>dep[y])   swap(x,y);
    for(int i=20; i>=0; i--) if(dep[f[y][i]]>=dep[x]) y=f[y][i];
    for(int i=20; i>=0; i--) if(f[x][i]!=f[y][i]) x=f[x][i],y=f[y][i];
    if(x!=y)  x=f[x][0];
    return x;
}
int juli(int a,int b){ return dis[a]+dis[b]-dis[lca(a,b)]*2;}
bool solve()
{
    int x,y,a,b,k;
    cin>>x>>y>>a>>b>>k;
    int lens=juli(a,b);
    if(lens<=k&&((lens&1)^(k&1))==0) return true;
    lens=juli(a,x)+juli(y,b)+1;
    if(lens<=k&&((lens&1)^(k&1))==0) return true;
    lens=juli(a,y)+juli(x,b)+1;
    if(lens<=k&&((lens&1)^(k&1))==0) return true;
    return false;
}
signed main()
{
    cin>>n;
    build();lca_init();
    int q;cin>>q;
    while(q--)
    {
        if(solve()) cout<<"YES\n";
        else cout<<"NO\n";
    }
    return 0;
}