B - 最近公共祖先·三 HihoCoder - 1069 rmq-lca

最新推荐文章于 2019-05-26 23:18:35 发布

ID_BePosit

最新推荐文章于 2019-05-26 23:18:35 发布

阅读量202

点赞数

CC 4.0 BY-SA版权

分类专栏： LCA

本文链接：https://blog.youkuaiyun.com/BePosit/article/details/83152474

LCA 专栏收录该内容

12 篇文章

订阅专栏

本文深入探讨了最近公共祖先（LCA）算法的实现细节，包括数据结构的设计、深度优先搜索（DFS）的运用以及跳跃指针技术。通过具体代码示例，详细解释了如何在树形结构中高效查找两个节点的最近公共祖先，适用于解决计算机科学中的多种问题。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

B - 最近公共祖先·三
HihoCoder - 1069

#include<bits/stdc++.h>
using namespace std;
#define maxn 110000
int n,k,q,sum;
int deep[110001],tot;
int head[110001],dp[110001][20];
map<string,int>mmp;
map<int,string>out;
string str1,str2;
struct node
{
    int v,to;
} edge[maxn*2];
void add(int u,int v)
{
    edge[++tot].v=v;
    edge[tot].to=head[u];
    head[u]=tot;
}
void dfs(int cur,int pre)
{
    deep[cur]=deep[pre]+1;
    dp[cur][0]=pre;
    for(int i=1; (1<<i)<=deep[cur]; i++)
        dp[cur][i]=dp[dp[cur][i-1]][i-1];
    for(int i=head[cur]; i!=-1; i=edge[i].to)
        if(edge[i].v!=pre)
            dfs(edge[i].v,cur);
}
int lca(int x,int y)
{
    if(deep[x]<deep[y])
        swap(x,y);
    for(int i=k; i>=0; i--)
        if(deep[x]-(1<<i)>=deep[y])
            x=dp[x][i];
    if(x==y)
        return x;
    for(int i=k; i>=0; i--)
        if(dp[x][i]!=dp[y][i])
        {
            x=dp[x][i];
            y=dp[y][i];
        }
    if(dp[x][0]==dp[y][0])
        return dp[x][0];
    else
        return -1;
}
int main()
{
    cin>>n;
    memset(head,-1,sizeof(head));
    for(int i=0; i<n; i++)
    {
        cin>>str1>>str2;
        if(mmp[str1]==0)
        {
            mmp[str1]=++sum;
            out[sum]=str1;
        }
        if(mmp[str2]==0)
        {
            mmp[str2]=++sum;
            out[sum]=str2;
        }
        add(mmp[str1],mmp[str2]);
    }
    k=log(sum)/log(2);
    dfs(1,0);
    cin>>q;
    while(q--)
    {
        cin>>str1>>str2;
        if(str1==str2)
            cout<<str2<<endl;
        else if(lca(mmp[str1],mmp[str2])==-1)
            cout<<-1<<endl;
        else
            cout<<out[lca(mmp[str1],mmp[str2])]<<endl;
    }
    return 0;
}