Lowest Common Ancestor of The Two Nodes in a Binary Tree

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;
                }
            }
        }
    }
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值