标签:最近公共祖先
root指向一棵二叉树的头结点,p和q分别指向该二叉树中任意两个结点的指针,编写算法,找到p和q的最近公共祖先结点r。
为了不失一般性,设p在q的左边。同时,为了编写方便,将以p和q所指结点的值代替p和q本身。
根据【后续遍历最后访问根节点,在递归算法中,根是压在栈底的】,很容易想到,采用后序遍历非递归算法:栈中存放二叉树结点的指针。当访问到某结点时,栈中所有的元素都是该结点的祖先。设一个辅助栈,当没有访问到p之前,辅助栈与原栈操作相同;当访问到p之后,不再对原栈操作,在操作辅助栈的过程中访问到q结束。最后将两栈从栈底元素逐个匹配,最后相同的结点就是最近公共祖先。
算法中设置一flag在记录是否访问到p。
TreeNode* ancestor(TreeNode* root, int x, int y){
if (root == NULL)
return NULL;
TreeNode* stack1[MaxSize];
TreeNode* stack2[MaxSize];
TreeNode* visited = NULL;
int top1 = -1, top2 = -1;
int flag = 0;
while (root || top1 != -1 || top2 != -1){
while (root && root->val != x && root->val != y){
if (flag == 0){
stack1[++top1] = root;
}
stack2[++top2] = root;
root = root->left;
}
if (root){
stack1[++top1] = root;
stack2[++top2] = root;
if (root->val == x){
flag = 1;
}
if (root->val == y){
int i = 0;
while (i <= top1&&i <= top2&&stack1[i] == stack2[i])
i++;
return stack1[--i];
}
root = root->left;
}
else{
root = stack2[top2];
if (!root->right || root->right == visited){
visited = root;
if (flag == 0)
top1--;
top2--;
root = NULL;
}
else{
root = root->right;
}
}
}
}
Lowest Common Ancestor of The Two Nodes in a Binary Tree
标签:最近公共祖先
原文地址:http://blog.csdn.net/kaitankedemao/article/details/44875571