hiho刷题日记——第十七天最近公共祖先·三

原创 2015年11月21日 17:20:58

问题

还是最近公共祖先问题。

思路

之前用了Tarjan算法来做。但是这里遇到一个问题就是如果只有一个询问,那么将很难决定到底是针对这个询问就直接进行计算还是等待一定数量的询问一起计算。毕竟无论是一个询问还是很多个询问,使用离线算法都是只需要做一次深度优先搜索就可以了的。
所以这一次改用一个在线算法。
先根据输入建立一棵家族树。
建立好家族树之后,深度遍历该树。使用一个数组来记录走过的路径(无论是从父节点进入还是子结点返回都要记录)。
然后这个问题就变为了一个RMQ问题。我们就可以使用RMQ-ST算法来解决了。

代码

#include<cstdio>
#include<iostream> 
#include<cstring>
#include<cmath>
#include<map>
using namespace std;

const int MAX_SIZE=100000+5;
int N;
int head[MAX_SIZE],dep[MAX_SIZE],last[MAX_SIZE],e=0;
int rmq[20][MAX_SIZE<<1],rmq_i=0;
int Log2[MAX_SIZE<<1];
map<string,int> str_id;
map<int,string> id_str;

struct EDGE
{
    int v,next;
}edge[MAX_SIZE];

void addEdge(int x,int y)
{
    edge[e].v=y;
    edge[e].next=head[x];
    head[x]=e++;
}

void dfs(int cur,int dp)
{
    dep[cur]=dp;
    last[cur]=rmq_i;//这句话之前没写。查错查了我一天。哭死。
    rmq[0][rmq_i++]=cur;
    int p=head[cur];
    while(p!=-1)
    {
        dfs(edge[p].v,dp+1);
        last[cur]=rmq_i;
        rmq[0][rmq_i++]=cur;
        p=edge[p].next;
    }

}

int main()
{
    memset(head,-1,sizeof(head));
    int id=0;
    string fn,sn;
    scanf("%d",&N);
    while(N--)
    {
        cin>>fn>>sn;
        if(!str_id.count(fn)) 
        {
            str_id[fn]=id;
            id_str[id]=fn;
            id++;
        }
        if(!str_id.count(sn))
        {
            str_id[sn]=id;
            id_str[id]=sn;
            id++;
        }
        addEdge(str_id[fn],str_id[sn]);
    } 
    dfs(0,0);

    int t=2,v=0;
    Log2[0]= Log2[1]=0;
    for(register int i=1;i<rmq_i;i++)   
    {
        if(i>t) {t<<=1;v++;}
        Log2[i]=v;
    }

    t=1;
    for(register int len=1;len<=Log2[rmq_i-1];len++)
    {
        for(int l=0;l+t<rmq_i;l++)
        {
            if(dep[rmq[len-1][l]]<dep[rmq[len-1][l+t]])
                rmq[len][l]=rmq[len-1][l];
            else rmq[len][l]=rmq[len-1][l+t];
        }
        t<<=1;
    }

    scanf("%d",&N);
    int l,r,len;
    while(N--)
    {
        cin>>fn>>sn;
        if(last[str_id[fn]]>last[str_id[sn]])
        {
            l=last[str_id[sn]];
            r=last[str_id[fn]];
        }
        else
        {
            l=last[str_id[fn]];
            r=last[str_id[sn]];
        }

        len=Log2[r+1-l];
        if(dep[rmq[len][l]]>dep[rmq[len][r+1-(1<<len)]])
            cout<<id_str[rmq[len][r+1-(1<<len)]]<<endl;
        else cout<<id_str[rmq[len][l]]<<endl;
    }
    return 0; 
}

二叉树最近公共祖先问题(O(n) time 且只遍历一遍,O(1) Space (不考虑函数调用栈的空间))

Tarjan算法很精妙,但是使用了并查集,需要额外O(n)的存储空间。上面博客中给的第三个方法也是需要记录根到节点的路径,需要O(log n)空间,当然考虑到一般情况下我们遍历树都是递归的方式,所以本...
  • smartxxyx
  • smartxxyx
  • 2015年04月02日 11:00
  • 5388

最近公共祖先(Least Common Ancestors,LCA)问题详解

问题描述 求有根树的任意两个节点的最近公共祖先。 最近公共祖先简称LCA(Lowest Common Ancestor),所谓LCA,是当给定一个有根树T时,对于任意两个结点u、v,找到一个离根最远的...
  • ywcpig
  • ywcpig
  • 2016年08月28日 01:45
  • 1275

在二叉树中找到两个节点的最近公共祖先

程序员代码面试指南(左程云)读书笔记 第三章 在二叉树中找到两个节点的最近公共祖先 题目:       给定一棵二叉树的头节点head,以及这棵二叉树的两个节点o1和o2,请返回o1和o2的...
  • qq_33323162
  • qq_33323162
  • 2016年08月30日 17:21
  • 2039

hiho刷题日记——第十三天最近公共祖先·一

题目已知一棵族谱树。查找两个人的最近公共祖先。输入每个测试点(输入文件)有且仅有一组测试数据。 每组测试数据的第1行为一个整数N,意义如前文所述。 每组测试数据的第2~N+1行,每行分别描述一对父...
  • AngcFun
  • AngcFun
  • 2015年11月14日 16:17
  • 264

hiho一下 第十五周——最近公共祖先·二(Trajan,离线LCA)

题目连接http://hihocoder.com/problemset/problem/1067题目大意就是一棵树求任意两个节点的最近公共祖先。算法描述在题目的提示里面有比较详细的解释。这里就不多说了...
  • tyh24689
  • tyh24689
  • 2015年03月09日 13:21
  • 736

[hiho]#1069 : 最近公共祖先·三 线段树|树转数组

描述:(原题地址:http://hihocoder.com/problemset/problem/1069?sid=601396) 给定一颗树,给出树根,以及一些查询pair,要求输出每条查询pai...
  • makeway123
  • makeway123
  • 2015年10月06日 20:36
  • 331

HIHO #1069 : 最近公共祖先·三(RMQ+DFS LCA在线算法)

题目链接 RMQ+DFS 实现在线的算法,dfs把树离散成一棵树,然后跑rmq,这个hihoCoder里面讲解的很好 #include #include #include #include #inc...
  • u013167299
  • u013167299
  • 2016年08月09日 21:52
  • 149

[hiho]#1067 : 最近公共祖先·二 离线算法

描述:(原题地址:http://hihocoder.com/problemset/problem/1067?sid=601284) 给定一颗树,给出树根,以及一些查询pair,要求输出每条查询pair...
  • makeway123
  • makeway123
  • 2015年10月06日 18:19
  • 387

HIHO #1067 : 最近公共祖先·二(并查集+dfs LCA离线算法)

题目链接l LCA 的算法还是有很多的,这里是并查集+dfs,是离线的算法 参考题目的提示,以及这个可以很好的理解 #include #include #include #include #in...
  • u013167299
  • u013167299
  • 2016年08月09日 19:19
  • 136

hiho#1062 : 最近公共祖先·一

#1062 : 最近公共祖先·一 时间限制:10000ms 单点时限:1000ms 内存限制:256MB 描述 小Ho最近发现了一个神奇的网站!虽然还...
  • su20145104009
  • su20145104009
  • 2016年05月18日 21:17
  • 3890
内容举报
返回顶部
收藏助手
不良信息举报
您举报文章:hiho刷题日记——第十七天最近公共祖先·三
举报原因:
原因补充:

(最多只允许输入30个字)