二叉树的下一个结点

题目描述

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

输入节点为A

如果A有右子树，遍历A右子树的左子树到底

如果A没有右子树，访问其父节点B，如果B的左子树是A输出返回，如果B的右子树是A继续遍历父节点，直到到达根节点

using namespace std;
/*
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)  
        {  
            TreeLinkNode* pout=NULL;
            if(pNode==NULL)  
                return NULL;  
            if(pNode->right!=NULL)  
            {  
                TreeLinkNode *pRight = pNode->right;  
                while(pRight->left!=NULL)  
                    pRight = pRight->left;  
               // pout=pRight;
                return pRight;  
            }  
            else if(pNode->next!=NULL)  
            {  
                TreeLinkNode *pCurNode = pNode;  
                TreeLinkNode *pParent = pNode->next;  
                while(pParent!=NULL && pCurNode == pParent->right)  
                {  
                    pCurNode = pParent;  
                    pParent = pParent->next;  
                }  
              //  pout=pParent;
                return pParent;
            } 
            return NULL;
        }  
};