二叉树的下一个结点

题目描述

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

首先给出一棵简单的二叉树如下所示

（1）如果这个结点有右子树，例如 2，那么它的下一个结点就是它右子树的最左结点 5

（2）如果这个结点没有右子树且为父结点的左结点，例如 7，那么它的下一个结点就是它父结点 ；如果这个结点没有右子树且为父结点的右结点，例如 4，就需要往指向父结点的指针向上遍历，直到找到一个是它父结点的左结点的结点 2，,要找的下一个结点就是所找到结点 2 的父结点 1

（3）如果这个结点既没有右子树，也没有左子树，那么它的下一个结点为空

代码如下：

/*
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 == nullptr)
            return nullptr;
        
        TreeLinkNode *pNext = nullptr;

        // 右子树不为空
        if (pNode->right != nullptr)
        {
            TreeLinkNode *pRight = pNode->right;
            while (pRight->left != nullptr)
            {
                pRight = pRight->left;
            }
            pNext = pRight;
        }

        // 右子树为空，父结点不为空
        else if (pNode->next != nullptr)
        {
            TreeLinkNode *pCur = pNode;
            TreeLinkNode *pParent = pNode->next;
            // 寻找一个是它父结点的左结点的结点
            while (pParent != nullptr && pParent->right == pCur)
            {
                pCur = pParent;
                pParent = pParent->next;
            }
            // 这个结点就是要找的结点
            pNext = pParent;
        }
        
        // 右子树为空，父结点为空
        return pNext;
    }
};