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; 
}
版权声明:本文为博主原创文章,未经博主允许不得转载。

相关文章推荐

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

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

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

描述:(原题地址:http://hihocoder.com/problemset/problem/1069?sid=601396) 给定一颗树,给出树根,以及一些查询pair,要求输出每条查询pai...

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

描述:(原题地址:http://hihocoder.com/problemset/problem/1067?sid=601284) 给定一颗树,给出树根,以及一些查询pair,要求输出每条查询pair...

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

#1062 : 最近公共祖先·一 时间限制:10000ms 单点时限:1000ms 内存限制:256MB 描述 小Ho最近发现了一个神奇的网站!虽然还...

hiho 17 最近公共祖先(三) 转换为区间最值查找 ST

问题描述描述上上回说到,小Hi和小Ho使用了Tarjan算法来优化了他们的“最近公共祖先”网站,但是很快这样一个离线算法就出现了问题:如果只有一个人提出了询问,那么小Hi和小Ho很难决定到底是针对这个...

hiho刷题日记——第九天状态压缩·二

hiho刷题日记——第九天天状态压缩·二题目小Hi和小Ho领到了一个大小为N*M的长方形盘子,他们可以用这个盒子来装一些大小为2*1的蛋糕。但是根据要求,他们一定要将这个盘子装的满满的,一点缝隙也不能...

hiho刷题日记——第三十一天小Hi小Ho的惊天大作战:扫雷·二

题目这次的扫雷是二维的了。根据三种规则做一次判断,判断出一定是雷和一定不是雷的数量。 1.如果某一个被探明的格子里所标的数字为0,那么它相邻的8个格子里的未探明格子被认作是一定不是地雷的格子。 2...

hiho刷题日记——第三十天小Hi小Ho的惊天大作战:扫雷·一

题目简化版扫雷,只有一行。 给出一行的周围3格中是雷的数量。 判断哪些一定是雷,一定不是雷。 思路因为给出的信息是环环相扣的,不用一个一个去试。 只需要试第一格是否为雷就可以通过数据算出后面是...

hiho刷题日记——第八天状态压缩·一

当状态转移方程中状态过多时,进行压缩

hihoCoder_#1069_最近公共祖先·三(RMQ-ST模板)

分析:LCA的在线算法,RMQ-ST算法。留存作为模板。 从树的根节点开始进行深度优先搜索,每次经过某一个点——无论是从它的父亲节点进入这个点,还是从它的儿子节点返回这个点,都按顺序记录下来。这样,...
内容举报
返回顶部
收藏助手
不良信息举报
您举报文章:深度学习:神经网络中的前向传播和反向传播算法推导
举报原因:
原因补充:

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