我lca写错了吗,为什么wa0
查看原帖
我lca写错了吗,为什么wa0
120340
lc_lca楼主2020/5/14 23:30
#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxn=100010;
int head[maxn],tot;
struct node{
    int v,nxt;
}edge[maxn<<1];
void addedge(int u,int v)
{
    edge[tot].v=v;
    edge[tot].nxt=head[u];
    head[u]=tot++;
}
void init()
{
    memset(head,-1,sizeof head);
    tot=0;
}
int n,q;
int f[50][2];
int dep[maxn];
void dfs(int u,int pre)
{
    dep[u]=dep[pre]+1;
    //cout<<" "<<u<<" "<<dep[u]<<endl;
    f[u][0]=pre;
    for(int i=1;i<=20;i++)
    {
        f[u][i]=f[f[u][i-1]][i-1];
    }
    for(int i=head[u];i!=-1;i=edge[i].nxt)
    {
        int v=edge[i].v;
        if(v==pre) continue;
        dfs(v,u);
    }
}
int lca(int u,int v)
{
    //if(u==v) return u;
    if(dep[u]<dep[v]) swap(u,v);
    for(int i=20;i>=0;i--)
    {
        if(dep[f[u][i]]>=dep[v]) u=f[u][i];
    }
    if(u==v) return u;
    for(int i=20;i>=0;i--)
    {
        if(dep[f[u][i]]!=dep[f[v][i]])
        {
            u=f[u][i];
            v=f[v][i];
        }
    }
    return f[u][0];
}
int dis(int u,int v)
{
    int Lca=lca(u,v);
    return dep[u]+dep[v]-2*dep[Lca];
}
int main()
{
    cin>>n>>q;
    init();
    for(int i=1;i<=n-1;i++)
    {
        int u,v;
        cin>>u>>v;
        addedge(u,v);
        addedge(v,u);
    }
    dep[0]=-1;
    dfs(1,0);
    //cout<<lca(1,1)<<endl;
    while(q--)
    {
        int a,b,c,d;
        cin>>a>>b>>c>>d;
        int e=lca(a,b),f=lca(c,d);
        if(dis(c,d)==dis(c,e)+dis(e,d) || dis(a,b)==dis(a,f)+dis(f,b))
        {
            cout<<"Y"<<endl;
        }
        else cout<<"N"<<endl;
    }
    return 0;
}

tks

2020/5/14 23:30
加载中...