1143 Lowest Common Ancestor (30 分)

1143 Lowest Common Ancestor (30 分)

题目大意

给出一棵搜索树的先序序列，问结点u和v的共同祖先是谁。

基本思路

数据结构：
定义哈希表map<int, bool> mp，用来标记树中所有出现过的结点值。
定义数组pre，用来存放先序序列。
基本思路：
因为是先序序列，是按照根、左子树、右子树的顺序；因为是搜索树，所以中序序列有序递增。
因为给出的是先序序列，那么就不妨遍历一遍先序序列pre，记值a为当前遍历到的结点值（当前子树的根结点）：如果结点u、v在以结点a为根结点的左子树或右子树上，那么u、v会大于a 或者 u、v小于a；只有当结点a等于u/v或者结点a在结点u、v之间时，因为是先序遍历的原因，以结点a为根结点的子树还没有开始遍历，所以该结点a就是我们要找的最近祖先结点，退出循环。

代码

具体数据结构和详细思路请看代码注释：

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<vector>
#include<map>
using namespace std;
const int maxn=10010;
vector<int> pre;//先序序列
map<int,bool> mp;//键是结点值，值是某个结点值是否出现过
int main(){
    int m,n,u,v,a;
    //读入测试样例组数、顶点个数、先序序列，给哈希表mp赋值
    cin>>m>>n;
    pre.resize(n);
    for(int i=0;i<n;i++){
        cin>>pre[i];
        mp[pre[i]]=true;
    }
    //读入每一组测试样例进行判断，输出相应的结果
    //二叉排序树，比根结点值小的结点在左子树，比根结点值大的结点在右子树
    //先序遍历的顺序是根、左子树、右子树，先序序列的顺序也是根，左子树，右子树
    /*按顺序遍历先序序列pre的每一个元素，遍历到的元素记为当前结点a：
      如果当前结点a不等于u/v，也不位于u、v之间：
         u、v可能在以当前结点a为根结点的左、右子树的某一个中，因为二叉搜索树性质，结点a会大于/小于u、v
         u、v可能不在以当前结点a为根结点的子树中，因为二叉搜索树和先序遍历的性质，当前结点a一定小于u、v
      如果当前结点a等于u/v 或者 位于u、v之间，因为二叉搜索树和先序遍历的性质，这时候根结点a的左右子树还没开始遍历，所以结点a就是最近的公共祖先结点，退出循环
      根据结点a和u、v的大小关系做出相应的输出
    */
    for(int i=0;i<m;i++){
        cin>>u>>v;
        for(int j=0;j<n;j++){
            a=pre[j];
            if((a==u)||(a==v)||(a>u&&a<v)||(a>v&&a<u)) break;
        }
        if(mp[u]==false&&mp[v]==false){
            printf("ERROR: %d and %d are not found.\n",u,v);
        }else if(mp[u]==false||mp[v]==false){
            printf("ERROR: %d is not found.\n",mp[u]==false?u:v);
        }else if(a==u||a==v){
            printf("%d is an ancestor of %d.\n",a,a==u?v:u);
        }else{
            printf("LCA of %d and %d is %d.\n",u,v,a);
        }
    }
    return 0;
}