剑指offer——06二叉树的下一个节点

题目描述

给定一个二叉树和其中的一个结点，请找出中序遍历顺序的下一个结点并且返回。注意，树中的结点不仅包含左右子结点，同时包含指向父结点的指针。

 

题目的意思是，在一颗二叉树的中序遍历中，给出其中一个节点，让你找到中序中该节点的下一个节点

 

题解：

情况 1：某节点有右子树，下一节点为右子树中的最左子节点

情况 2：某节点无右子树，且就是他父节点左子节，则下一节点为父节点

情况 3：某节点无右子树，且为他父节点的右子节点，则向上递归寻找它的父节点，直到根节点或某个父节点是它自身父节点的左子节点

 

class Solution {
public:
    TreeLinkNode* GetNext(TreeLinkNode* pNode)
    {
        if (pNode == nullptr)
            return nullptr;
        if (pNode->right != nullptr)//有右子树，则为右子树的最左节点
        {
            pNode = pNode->right;
            while (pNode->left != nullptr)
                pNode = pNode->left;
            return pNode;
        }
        else if (pNode->right == nullptr)
        {
            while (pNode->next!=nullptr && pNode != pNode->next->left)
                pNode = pNode->next;
            return pNode->next;
        }
        return nullptr;
    }
};