LCA最近公共祖先(RMQ、Tarjan)

题目:http://poj.org/problem?id=1330

讲解:http://blog.csdn.net/tingyun_say/article/details/52092464

离线算法:dfs+并查集(Tarjan)

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<vector>
using namespace std;
const int N=1e4+107;
const int inf=0x3f3f3f3f;
int n,q1,q2;
int f[N],a[N];
bool c[N],flag[N];
vector<int> v[N];
int find(int k)
{
    if(f[k]==k) return k;
    f[k]=find(f[k]);
    return f[k];
}
int unions(int x,int y)
{
    int xx=find(x),yy=find(y);
    f[yy]=xx;
    return 0;
}
int tarjan(int u)
{
    for(int i=0;i<v[u].size();i++)
    {
        tarjan(v[u][i]);
        unions(u,v[u][i]);
        a[find(u)]=u;
    }
    c[u]=1;
    if(q1==u&&c[q2]) printf("%d\n",a[find(q2)]);
    else if(q2==u&&c[q1]) printf("%d\n",a[find(q1)]);
    return 0;
}
int main()
{
    int t,i;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d",&n);
        for(i=1;i<=n;i++)
        {
            v[i].clear();
            f[i]=i;
            a[i]=0;
            c[i]=0;
            flag[i]=1;
        }
        for(i=1;i<n;i++)
        {
            int x,y;
            scanf("%d%d",&x,&y);
            flag[y]=0;
            v[x].push_back(y);
        }
        scanf("%d%d",&q1,&q2);
        for(i=1;i<=n;i++) if(flag[i]) break;
        tarjan(i);
    }
    return 0;
}
在线算法:dfs+RMQ(RMQ)
#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<vector>
using namespace std;
const int N=1e4+107;
const int inf=0x3f3f3f3f;
const int log=20;
int dp[N][log],depth[N],deg[N];
struct node{
    int to;
    node *next;
}g[2*N],*cur,*head[N];
void addnode(int u,int v)
{
    cur->to=v;
    cur->next=head[u];
    head[u]=cur++;
}
void dfs(int u)
{
    depth[u]=depth[dp[u][0]]+1;
    for(int i=1;i<log;i++) dp[u][i]=dp[dp[u][i-1]][i-1];
    for(node *it=head[u];it;it=it->next) dfs(it->to);
}
int lca(int u,int v)
{
    if(depth[u]<depth[v]) swap(u,v);
    for(int st=1<<(log-1),i=log-1;  i>=0   ;i--,st>>=1)
        if(st<=depth[u]-depth[v]) u=dp[u][i];
    if(u==v) return u;
    for(int i=log-1;i>=0;i--)
        if(dp[v][i]!=dp[u][i])
        v=dp[v][i],u=dp[u][i];
    return dp[u][0];
}
void init(int n)
{
    for(int i=0;i<=n;i++)
    {
        dp[i][0]=0;
        head[i]=NULL;
        deg[i]=0;
    }
    cur=g;
}
int main()
{
    int t;
    scanf("%d",&t);
    while(t--)
    {
        int n,u,v;
        scanf("%d",&n);
        init(n);
        for(int i=0;i<n-1;i++)
        {
            scanf("%d%d",&u,&v);
            addnode(u,v);
            deg[v]++;
            dp[v][0]=u;
        }
        for(int i=1;i<=n;i++)
            if(deg[i]==0){dfs(i);break;}
        scanf("%d%d",&u,&v);
        printf("%d\n",lca(u,v));
    }
}
HDU  2586
Lca
#define N 50005
vector<int> mp[N],w[N],query[N],num[N];
int p[N],d[N],res[N];
bool vis[N];
int n;

void Init()
{
    int i;
    for(i=1;i<=n;i++)
    {
        mp[i].clear();
        w[i].clear();
        query[i].clear();
        num[i].clear();
        p[i]=i;
        d[i]=0;
        vis[i]=false;
    }
}

int Find(int x)
{
    if(p[x]!=x)
        p[x]=Find(p[x]);
    return p[x];
}

void Union(int x,int y)
{
    x=Find(x);
    y=Find(y);
    if(x!=y)
        p[y]=x;
}

void Tarjan(int cur,int v)
{
    int size,i,tmp;

    vis[cur]=true;
    d[cur]=v;
    size=mp[cur].size();
    for(i=0;i<size;i++)
    {
        tmp=mp[cur][i];
        if(vis[tmp]) continue;
        Tarjan(tmp,v+w[cur][i]);
        Union(cur,tmp);
    }
    size=query[cur].size();
    for(i=0;i<size;i++)
    {
        tmp=query[cur][i];
        if(!vis[tmp])continue;
        res[num[cur][i]]=d[cur]+d[tmp]-2*d[Find(tmp)];
    }
}

int main()
{
    int T,q,a,b,c,i;

    scanf("%d",&T);
    while(T--)
    {
        scanf("%d%d",&n,&q);
        Init();
        for(i=1;i<n;i++)
        {
            scanf("%d%d%d",&a,&b,&c);
            mp[a].push_back(b);
            w[a].push_back(c);
            mp[b].push_back(a);
            w[b].push_back(c);
        }
        for(i=0;i<q;i++)
        {
            scanf("%d%d",&a,&b);
            query[a].push_back(b);
            query[b].push_back(a);
            num[a].push_back(i);
            num[b].push_back(i);
        }
        Tarjan(1,0);
        for(i=0;i<q;i++)
            printf("%d\n",res[i]);
    }
    return 0;
}


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
Tarjan算法是一种用于求解最近公共祖(Least Common Ancestors,LCA)问题的离线算法。算法的核心思想是利用深度优先搜索(DFS)和并查集(Union Find)来解决问题。 首先,我们从根节点开始遍历每一个节点,并将节点分为三类,用st[]数组表示。0代表还未被遍历,1代表正在遍历这个点,2代表已经遍历完这个点并且回溯回来了。这样的划分有助于确定节点的最近公共祖先。 在Tarjan算法中,我们一边遍历一边回应查询。每当遍历到一个节点时,我们查找与该节点相关的所有查询。如果查询中的节点已经被遍历完(即st[]值为2),我们可以利用已经计算好的信息来计算它们的最近公共祖先最近公共祖先的距离可以通过两个节点到根节点的距离之和减去最近公共祖先节点到根节点的距离来计算。 在Tarjan算法中,我们可以通过深度优先搜索来计算dist[]数组,该数组表示每个节点到根节点的距离。我们可以利用父节点到根节点的距离加上边的权值来计算每个节点到根节点的距离。 最后,我们可以通过并查集来操作st[]数组。当遍历完一个节点的所有子树后,将子树中的节点放入该节点所在的集合。这样,每个子树的节点的最近公共祖先都是该节点。 综上所述,Tarjan算法利用DFS和并查集来求解最近公共祖先问题。它的时间复杂度为O(n+m),其中n是节点数,m是查询次数。通过该算法,我们可以高效地解决最近公共祖先的问题。<span class="em">1</span><span class="em">2</span><span class="em">3</span> #### 引用[.reference_title] - *1* [最近公共祖先tarjan](https://blog.csdn.net/qq_63092029/article/details/127737575)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v92^chatsearchT3_1"}}] [.reference_item style="max-width: 33.333333333333336%"] - *2* [【模版】Tarjan离线算法求最近公共祖先(LCA)](https://blog.csdn.net/weixin_43359312/article/details/100823178)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v92^chatsearchT3_1"}}] [.reference_item style="max-width: 33.333333333333336%"] - *3* [Tarjan算法求解最近公共祖先问题](https://blog.csdn.net/Yeluorag/article/details/48223375)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v92^chatsearchT3_1"}}] [.reference_item style="max-width: 33.333333333333336%"] [ .reference_list ]

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值