二叉树的下一个节点-牛客题解

最新推荐文章于 2022-08-18 00:00:00 发布

Zero_Clear

最新推荐文章于 2022-08-18 00:00:00 发布

阅读量197

点赞数

分类专栏：算法剑指offer

本文链接：https://blog.csdn.net/u011932817/article/details/104221335

版权

算法同时被 2 个专栏收录

12 篇文章 0 订阅

订阅专栏

剑指offer

1 篇文章 0 订阅

订阅专栏

环境

C++ 11(clang3.9)

思路推倒过程

前提: 要求是中序遍历的下一节点
- 中序: 先左子树再根再右子树

通过代码

class Solution
{
public:
    TreeLinkNode *GetNext(TreeLinkNode *pNode)
    {
        //思路
        //
        // - 下一节点为当前节点的右子树的最左子节点
        // - 当没有右子树时 下一节点为 向上第一个左树枝的父节点

        
        if (pNode == nullptr)
            return nullptr;
        TreeLinkNode *next = nullptr;
        //当前节点没有右子树
        if (pNode->right == nullptr)
        {
            TreeLinkNode *pCur = pNode;
            TreeLinkNode *pParent = pCur->next;
            while (pParent != nullptr && pParent->right == pCur)
            {
                pCur = pParent;
                pParent = pParent->next;
            }
            next = pParent;
        }
        else //当前节点有右子树
        {
            TreeLinkNode *pRight = pNode->right;
            while (pRight->left != nullptr)
            {
                pRight = pRight->left;
            }
            next = pRight;
        }
        return next;
    }

};