剑指offer第57题:二叉树的下一个结点

• 题目描述
• 二叉树的下一个结点
• 源码

题目描述

二叉树的下一个结点

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

源码

/*
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)//1.若该点存在右子树，则返回右子树最左边的节点。
     {
         TreeLinkNode* Temp = pNode->right;
         while(Temp->left!=NULL)
         {
             Temp = Temp->left;
         }
         return Temp;
     }
    while(pNode->next!=NULL)//2.当该点不为根节点：a.若该点是左儿子，则返回其父节点，若为右儿子，则其父节点已经遍历过，向上遍历直到找到作为左儿子的父辈节点，返回该节点的父节点。
    {
       TreeLinkNode* proot = pNode->next;
       if(pNode == proot->left)
       {
           return proot;
       }
       pNode = pNode->next;
    }
    return NULL;//3.跳出2中循环说明已经遍历到了根节点还未找到，说明最初的pNode在树的最右端，无下一个节点。
   }
};