题目链接:https://pintia.cn/problem-sets/994805342720868352/problems/994805343727501312
题目大意: 给一个二叉排序树的前序遍历, 给M (≤ 1,000)次查询,每次查询给出两个节点。查询他们的最低公共祖先。
分析:这道题给出来的树是一棵二叉排序树
二叉排序树又称“二叉查找树”、“二叉搜索树”。
二叉排序树:或者是一棵空树,或者是具有下列性质的二叉树:
- 若它的左子树不空,则左子树上所有结点的值均小于它的根结点的值;
- 若它的右子树不空,则右子树上所有结点的值均大于它的根结点的值;
- 它的左、右子树也分别为二叉排序树。
对于二叉排序树来说,这道题就有了比较特殊的解法。前序遍历是先根节点 然后再左子树和右子树,对于一个节点来说,如果大于u并且小于v的话,那么u一定是在该节点的左子树上,v一定是在该节点的右子树上。所以对于前序遍历来说。只要找到的第一个节点满足条件的节点,就是要找的答案。
#include<iostream>
#include<map>
#include<cstdio>
using namespace std;
const int MAXN = 1e4+10;
int pre[MAXN];
map<int,int> m;
int main()
{
int M,N;
scanf("%d%d",&M,&N);
for(int i = 0 ; i < N ; i++)
{
scanf("%d",&pre[i]);
m[pre[i]]++;
}
while(M--)
{
int u,v;
scanf("%d%d",&u,&v);
if(m[u]==0&&m[v]){printf("ERROR: %d is not found.\n",u);continue;}
if(m[v]==0&&m[u]){printf("ERROR: %d is not found.\n",v);continue;}
if(!m[v]&&!m[u]){printf("ERROR: %d and %d are not found.\n",u,v);continue;}
for(int i = 0 ; i < N ; i++)
{
if( (pre[i]>=u&&pre[i]<=v) || (pre[i]>=v&&pre[i]<=u) )
{
if(pre[i]==u){printf("%d is an ancestor of %d.\n",u,v);break;}
else if(pre[i]==v){printf("%d is an ancestor of %d.\n",v,u);break;}
else{printf("LCA of %d and %d is %d.\n",u,v,pre[i]);break;}
}
}
}
return 0;
}