PAT甲级1151 LCA in a Binary Tree (30 分)（最近公共祖先问题）

林空鹿饮溪cyd

于 2021-03-09 16:09:47 发布

阅读量187

点赞数

本文链接：https://blog.csdn.net/CSDN_Ysu/article/details/114585042

版权

PAT甲级1151 LCA in a Binary Tree (30 分)（最近公共祖先问题）（参考柳神）

题意：给出一棵树的中序和先序序列，再给出两个点，给这两个点最近的公共祖先
思路：
自己写代码的时候觉得就是模板题吧，构建出来二叉树之后用BFS增加结点的父亲属性，接下去比较这两个点的父亲节点就可以了。结果超时了…同时也没有注意到用map，只用的一个数组，有段错误。
自己的代码就不贴了，参考此博客，并没有建树，而是直接用LCA算法写的
step1：若a，b在根节点左边，则最近公共祖先在这个根结点的左边查找，若a，b在根节点右边，则最近公共祖先在这个根结点的右边查找，若a，b在根结点两边，则最近公共祖先就是根结点。不断调用递归：代码如下：

#include<bits/stdc++.h>
using namespace std;
map<int,int> pos;
vector<int>in,pre;
void lca(int inl,int inr,int preRoot,int a,int b)
{
    if(inl>inr) return;
    int inRoot=pos[pre[preRoot]],aIn=pos[a],bIn=pos[b];//找到a,b的数组下标，和此时的根结点数组下标地址
    if(aIn<inRoot&&bIn<inRoot)
        lca(inl,inRoot-1,preRoot+1,a,b);
    else if((aIn>inRoot&&bIn<inRoot)||(aIn<inRoot&&bIn>inRoot))
        printf("LCA of %d and %d is %d.\n", a, b, in[inRoot]);
    else if(aIn>inRoot&&bIn>inRoot)
        lca(inRoot+1,inr,preRoot+(inRoot-inl)+1,a,b);
    else if(aIn==inRoot)
        printf("%d is an ancestor of %d.\n", a, b);
    else if (bIn == inRoot)
            printf("%d is an ancestor of %d.\n", b, a);
}

int main()
{
    int m,n,a,b;
    scanf("%d%d",&m,&n);
    in.resize(n+1),pre.resize(n+1);
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&in[i]);
        pos[in[i]]=i;
    }
    for(int i=1;i<=n;i++)scanf("%d",&pre[i]);
    for(int i=0;i<m;i++)
    {
        scanf("%d%d",&a,&b);
        if(pos[a]==0&&pos[b]==0)
            printf("ERROR: %d and %d are not found.\n", a, b);
        else if (pos[a] == 0 || pos[b] == 0)
            printf("ERROR: %d is not found.\n", pos[a] == 0 ? a : b);
        else
            lca(1, n, 1, a, b);

    }
    return 0;
}