最近公共祖先(离线)

题目链接
题意:给你一棵树,后面是前面的父节点,k次查询,如果a是b的祖先输出1,b是a的祖先输出2,否则输出0;
思路:根据最近公共祖先判断得出结果;

/*LCA(离线算法)
主函数除建边外还应调用
init();
dir[1]=0;
tarjan(1);
*/
#include <stdio.h>
#include <string.h>
#include <iostream>
#include <algorithm>
using namespace std;
const int maxn=1e5+10;
 struct zp
 {
     int a,b;
 }node[maxn*4];
struct note
{
    int u,v,w,lca,next;
}edge[maxn*4],edge1[maxn*4];

int head[maxn*4],ip,head1[maxn*4],ip1;///需要建两次边。1,该树的边       2,需要查询的两点
int m,n;
int father[maxn*4],vis[maxn*4],dir[maxn*4];
///依次表示u点的祖先、标记是否访问过,到根节点的距离

void init()
{
    memset(vis,0,sizeof(vis));
    memset(dir,0,sizeof(dir));
    memset(head,-1,sizeof(head));
    memset(head1,-1,sizeof(head1));
    ip=ip1=0;
}

void addedge(int u,int v,int w)
{
    edge[ip].v=v,edge[ip].w=w,edge[ip].next=head[u],head[u]=ip++;
}

void addedge1(int u,int v)
{
    edge1[ip1].u=u,edge1[ip1].v=v,edge1[ip1].lca=-1,edge1[ip1].next=head1[u],head1[u]=ip1++;
}

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

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

void tarjan(int u)
{
    vis[u]=1;
    father[u]=u;
    for(int i=head[u];i!=-1;i=edge[i].next)
    {
        int v=edge[i].v;
        int w=edge[i].w;
        if(!vis[v])
        {
           dir[v]=dir[u]+w;
           tarjan(v);
           Union(u,v);
        }
    }
    for(int i=head1[u];i!=-1;i=edge1[i].next)
    {
        int v=edge1[i].v;
        if(vis[v])
        {
            edge1[i].lca=edge1[i^1].lca=father[Find(v)];
        }
    }
}
int main()
{
    int n;
    while(~scanf("%d",&n))
    {
        init();
        int root;
        for(int i=0;i<n;i++)//建树
        {
            int a,b;
            scanf("%d%d",&a,&b);
            if(b==-1)
                root=a;
            else
            addedge(b,a,1);
        }
        int k;
        scanf("%d",&k);
        for(int i=0;i<k;i++)//离线查询建边
        {
            int a,b;
            scanf("%d%d",&a,&b);
            node[i].a=a;
            node[i].b=b;
            addedge1(a,b);
            addedge1(b,a);
        }
        dir[root]=0;
        tarjan(root);
        for(int i=0;i<k*2;i+=2)
        {
            int a=node[i/2].a,b=node[i/2].b;
            int tmp=edge1[i].lca;//a和b的公共最近公共祖先
            if(tmp==a)
                printf("1\n");
            else if(tmp==b)
                printf("2\n");
            else
                printf("0\n");
        }
    }
    return 0;
}
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值