码迷,mamicode.com
首页 > 其他好文 > 详细

Lowest Common Ancestor of The Two Nodes in a Binary Tree

时间:2015-04-04 21:18:20      阅读:150      评论:0      收藏:0      [点我收藏+]

标签:最近公共祖先

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

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!