PAT-A 1143 LCA

PAT-A 1143 LCA

题意分析

这道题和1151其实基本上差不多，不需要去建立实际的二叉树。详细的思路分析过程可以参考柳神的博客：
柳婼

AC代码

#include<unordered_map>
#include<cstdio>
#include<iostream>
using namespace std;
unordered_map<int,int> pos_mapping;
const int MAX_N = 10010;
int pre[MAX_N];
int n,m,a,b;
//这个函数主要的目的在于，找到下标为k的结点的第一个左子树
/*
    这里利用了BST的性质。
    因为按照题意，BST是唯一的，所以它的构建过程就是按照给出的顺序不断地往树里面插入新的结点。
    只要了解了BST的性质，就能够明白，由于前面的所有点都已经被插进去了，所以最新的那个点它的子树一定是在它后面的。
*/
int ltree_root(int k){//查询下标为k的数据的左子树的根节点。
    for(int i = k+1;i<=n;i++){
        if(pre[i] < pre[k]){
            return i;
        }
    }
    return 0;//表示没有左子树
}
int rtree_root(int k)
{
    for(int i = k+1;i<=n;i++){
        if(pre[i] > pre[k]){
            return i;
        }
    }
    return 0;//表示没有右子树
}
//a、b表示两个数字的值，root_pos表示根节点的下标
void lca(int a,int b,int root_pos){
    if(root_pos == 0) return;//如果已经访问到不存在的点了，就直接退出.
    int root_val = pre[root_pos];//先求根节点的值
    //如果a和b的值分别位于根节点值的两侧，那么说明，现在的这个根节点就是a、b的最近公共祖先
    if((a < root_val && b > root_val) ||(a > root_val && b < root_val)){
        printf("LCA of %d and %d is %d.\n",a,b,root_val);
    }
    //如果a和b都位于根节点的左侧，那么就往左子树递归；
    else if(a < root_val && b <root_val){
        lca(a,b,ltree_root(root_pos));
    }
    //如果a和b都位于根节点的右侧，那么就往右子树递归。
    else if(a > root_val && b > root_val){
        lca(a,b,rtree_root(root_pos));
    }
    //如果a是根节点了，那么就说明a是b的祖先
    else if(a == root_val){
        printf("%d is an ancestor of %d.\n",a,b);
    }
    //同上
    else if(b == root_val){
        printf("%d is an ancestor of %d.\n",b,a);
    }
}

int main(){
    cin>>m>>n;
    for(int i = 1;i<=n;i++){
        scanf("%d",&pre[i]);
        pos_mapping[pre[i]] = i;
    }
    while (m--)
    {
        scanf("%d%d",&a,&b);
        if(pos_mapping[a] == 0 && pos_mapping[b] == 0){
            printf("ERROR: %d and %d are not found.\n",a,b);
        }
        else if(pos_mapping[a] == 0 || pos_mapping[b] == 0){
            printf("ERROR: %d is not found.\n",pos_mapping[a] == 0?a:b);
        }
        else{
            lca(a,b,1);
        }
    }
   //system("pause");
    return 0;
}