剑指offer57：二叉树的下一个结点

1 题目描述

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

2 思路和方法

　　中序遍历，下一个结点有两种情况

　　a. 当前结点有右子树，就找出右子树中的最左的结点；

　　b. 当前结点没有右子树  就往它的父节点找，找到第一个结点是它的父节点的左子节点的结点时停止，下一个结点就是该节点的父节点；

作如下表述：

（1）’：有右子树的，那么下个结点就是右子树最左边的点；

（2）’：没有右子树的，给出结点是父结点的左孩子，返回父结点；

（3）’：没有右子树，给出结点是父结点的右孩子，把给出结点的父节点作为下一个遍历的节点，向上回溯，直到当前结点是其父节点的左孩子时停止【直到找到一个父节点X，并且这个父节点X是其本身的父节点Y的左孩子为止】，下一个结点就是当前结点的父节点proot【return proot】。【TreeLinkNode *proot=pNode->next；proot->left==pNode，pNode是当前节点，pNode->next是当前节点的父节点，proot->left是其父结点的左结点】，

例子：　先序遍历：1 2 4 6 7 5 8 3（根左右）；　　中序遍历：6 4 7 2 5   3（左根右）【给出的结点为8，下一个结点为1为例】　　后序遍历：6 7 4 8 5 2 3 1（左右根）

3 C++核心代码

 /*
 struct TreeLinkNode {
     int val;
     struct TreeLinkNode *left;
     struct TreeLinkNode *right;
     struct TreeLinkNode *next;
     TreeLinkNode(int x) :val(x), left(NULL), right(NULL), next(NULL) {

     }
 };
 */
 class Solution {
 public:
     TreeLinkNode* GetNext(TreeLinkNode* pNode)
     {
         if(pNode==NULL)
             return NULL;
         if(pNode->right!=NULL)
         {
             pNode=pNode->right;
             while(pNode->left!=NULL)
                 pNode=pNode->left;           //有右子树的，那么下个结点就是右子树最左边的点
             return pNode;
         }
         while(pNode->next!=NULL)
         {                                        //pNode，是当前节点
             TreeLinkNode *proot=pNode->next;     //proot，当前节点的父节点
             if(proot->left==pNode)               //proot->left，当前节点的父节点的左结点
                 return proot;
             pNode=pNode->next;
         }
         return NULL;
     }
 };

参考资料

https://blog.csdn.net/langxue4516/article/details/82292216