二叉树的下一个节点

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

解题思路：

• 要找出中序遍历顺序的下一个节点，则要首先搞清楚中序遍历的顺序，即先遍历左子树再遍历根节点再遍历右子树
• 要找出下一个节点，分为下面几种情况：
• 情况一：给定节点有右孩子，那么下一个节点则为它的右子树的最左节点（不论该节点是否是根节点，都是一样的）
• 情况二：给定节点没有右孩子，且是它父亲节点的左孩子，那么下一个节点则为它的父亲节点
• 情况三：给定节点没有右孩子，在根节点的左子树且是它父亲节点的右孩子，那么就证明它是根节点左子树的最右节点，下一个要遍历的则为根节点
• 情况四：给定节点在根节点右子树，没有右孩子且是它父亲节点的右孩子节点，那么它是中序遍历的最后一个节点，返回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==NULL)
            return NULL;
        if(pNode->right!=NULL)
        {
            pNode=pNode->right;
            while(pNode->left!=NULL)
                pNode=pNode->left;
            return pNode;
        }
        while(pNode->next!=NULL)
        {
            TreeLinkNode* pRoot=pNode->next;
            if(pRoot->left==pNode)
                return pRoot;
            pNode=pNode->next;
        }
        return NULL;
    }
};