【每日一题】求二叉树中两个结点的最近公共祖先结点

求二叉树中两个结点的最近公共祖先结点

首先我们拿到题目是和二叉树相关,二叉树其中如果是搜索树的话那这个问题很简单,我们直接按照节点的大小比较就可以,如果根节点比两个节点值都大那两个节点的值都在左子树,以此类推。可是更常见的情况是一般二叉树。
对于一般二叉树而言,我们要找到最近的公共祖先节点,我第一个反应是把这两个节点的路径都找到,然后对比路径,然后找到最近的公共祖先节点。
1.找到两个节点路径
2.对比两个节点路径
我们先来分析第一步,我们一般实现二叉树的接口都是递归,想要拿到路径我们只能通过栈与回溯模拟递归,每一次递归到叶子节点要是没找到就让它置空,当左右都为空的时候就新出来一批叶子节点我们继续找,直到找到这个时候栈里边存的是路径。
然后进行第二步,我们拿到了两个栈里边存的是路径,取出其中一个的最后一个路径与第一个栈中的节点进行匹配,不匹配的话出栈然后继续取栈顶进行匹配,直到全部出栈说明没有公共祖先节点。
以下为代码实现

//求二叉树中两个结点的最近公共祖先结点
//这里为了方便测试直接用节点存的数据传参
BTNode* CommentParents(BTNode* root, int a, int b)
{
    Stack s1,s2;
    BTNode* root1 = root;
    BTNode* root2 = root;
    StackInit(&s1);
    StackInit(&s2);
    StackPush(&s1,root1);
    StackPush(&s2,root2);
    while (StackEmpty(&s1) == 1)
    {
        if (StackTop(&s1)->_data == a)
        {
            break;//找到了a并且栈里边存a的路径
        }
        else
        {
            if (StackTop(&s1)->left == NULL && StackTop(&s1)->right == NULL)
            {
                StackPop(&s1);//走到底了回溯
            }
            else
            {
                if (StackTop(&s1)->right == NULL)
                {
                    StackPush(&s1,StackTop(&s1)->left);//如果左子树不为空,左子树入栈
                }
                else
                {
                    StackPush(&s1, StackTop(&s1)->right);
                }
            }
        }
    }
    while (StackEmpty(&s2) == 1)
    {
        if (StackTop(&s2)->_data == b)
        {
            break;//找到了b并且栈里边存b的路径
        }
        else
        {
            if (StackTop(&s2)->left == NULL && StackTop(&s2)->right == NULL)
            {
                StackPop(&s2);//走到底了回溯
            }
            else
            {
                if (StackTop(&s2)->right == NULL)
                {
                    StackPush(&s2, StackTop(&s2)->left);//如果左子树不为空,左子树入栈
                }
                else
                {
                    StackPush(&s2, StackTop(&s2)->right);
                }
            }
        }
    }
    //走到这里s1s2分别存的ab的路径
    while (StackEmpty(&s1) + StackEmpty(&s2) == 2)
    {
        for (int i = s1._top;i ==0;i--)
        {
            if (s1._array[i]->_data == StackTop(&s2)->_data)
            {
                return s1._array[i];
            }
        }
        StackPop(&s2);
    }
    return NULL;
}

猜你喜欢

转载自blog.csdn.net/qq9116136/article/details/80521920