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

题目描述

 

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

 

解题思路

分情况考虑如下：

1. 若该节点为空，则直接返回NULL；
2. 若该节点有右子树，则它的下一个节点一定是右子树中最左边的叶子节点，所以从该节点右孩子开始循环，依次指向它的左孩子节点，直到左孩子节点为空停止，此时指针指向的节点即为目标节点；
3. 若该节点无右子树，且该节点为它父节点的左孩子，那么中序遍历的下一个节点就是它的父节点；
4. 若该节点无右子树，且该节点为它父节点的右孩子，此时可以沿着指向父节点的指针一直向上遍历，直到找到一个是它父节点的左子节点的节点。如果这样的节点存在，那么这个节点的父节点就是我们要找的下一个节点；
5. 若以上情况都不满足，则该节点下一个节点为NULL。

代码

 /*
 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)
             return NULL;
         else if(pNode->right){
             pNode=pNode->right;
             while(pNode->left)
                 pNode=pNode->left;
             return pNode;
         }
         else if(pNode->next&&pNode==pNode->next->left)
             return pNode->next;
         else if(pNode->next&&pNode==pNode->next->right){
             while(pNode->next&&pNode!=pNode->next->left)
                 pNode=pNode->next;
             return pNode->next;
         }
         else
             return NULL;
     }
 };