剑指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;
     }
 };