给定一棵二叉搜索树的先序遍历序列,要求你找出任意两结点的最近公共祖先结点(简称 LCA)。
Input
输入的第一行给出两个正整数:待查询的结点对数
M
M
M(
≤
1000
\le 1 000
≤1000)和二叉搜索树中结点个数
N
N
N(
≤
10000
\le 10 000
≤10000)。
随后一行给出
N
N
N 个不同的整数,为二叉搜索树的先序遍历序列。
最后
M
M
M 行,每行给出一对整数键值
U
U
U 和
V
V
V。
所有键值都在整型int范围内。
Output
对每一对给定的
U
U
U 和
V
V
V,如果找到 A 是它们的最近公共祖先结点的键值,则在一行中输出 LCA of U and V is A.。
但如果
U
U
U 和
V
V
V 中的一个结点是另一个结点的祖先,则在一行中输出 X is an ancestor of Y.,其中 X 是那个祖先结点的键值,Y 是另一个键值。
如果 二叉搜索树中找不到以
U
U
U 或
V
V
V 为键值的结点,则输出 ERROR: U is not found. 或者 ERROR: V is not found.,或者 ERROR: U and V are not found.。
难点在于建树
二叉搜索树性质:左儿子的值<根节点的值<右儿子的值,所以中序遍历是递增的序列。
题目给出了先序遍历,那么从小到大排序后就是中序遍历,建树的过程位根据先序遍历和中序遍历建立二叉树。
递归建树后,根据二叉搜索树的性质递归查找lca的值。
如果x,y都大于当前节点的值,说明x,y都在当前节点的右子树;如果x,y都小于当前节点的值,说明x,y都在当前节点的左子树;否则,说明x,y分列在当前节点的左、右子树,那么当前节点就是lca。
#include
using namespace std;
const int maxn=1e4+100;
#define debug(x) cout<<#x<<":"<<x<<endl;
struct node{
int val;
node *l;
node *r;
};
node tr[maxn];
int a[maxn],b[maxn],idx;
map<int,int>mp;
node* build(int len,int u,int st){
if(len<1) return 0;
node* now=&tr[idx++];///根节点
now->val=a[u];
int t=0;
t=lower_bound(b+t+st,b+len+st,a[u])-b-st;///找到根节点
/// debug(t);
now->l=build(t,u+1,st);
now->r=build(len-t-1,u+t+1,st+t+1);
return now;
}
int lca(int x,int y,node* root){
if(x>root->val&&y>root->val)
return lca(x,y,root->r);
else if(x<root->val&&y<root->val)
return lca(x,y,root->l);
else return root->val;
}
int main()
{
int m,n;cin>>m>>n;
for(int i=0;i<n;i++){
cin>>a[i];
///debug(a[i]);
b[i]=a[i];
mp[b[i]]=1;///标记出现过
}
sort(b,b+n);///中序遍历
node* root=build(n,0,0);
while(m--){
int x,y;
cin>>x>>y;
if(!mp.count(x)&&!mp.count(y)) printf("ERROR: %d and %d are not found.\n",x,y);
else if(!mp.count(x)&&mp.count(y)) printf("ERROR: %d is not found.\n",x);
else if(mp.count(x)&&!mp.count(y)) printf("ERROR: %d is not found.\n",y);
else{
int par=lca(x,y,root);
if(par==x){
printf("%d is an ancestor of %d.\n",x,y);
}
else if(par==y){
printf("%d is an ancestor of %d.\n",y,x);
}
else{
printf("LCA of %d and %d is %d.\n",x,y,par);
}
}
}
return 0;
}
欢迎分享,转载请注明来源:内存溢出
评论列表(0条)