编程--二叉树的下一个节点

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

/*
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;
        }
        //1.判断有没有右子树，pNode节点有右孩子,下一个结点就是以pNode结点的右孩子为根的子树中的最左下结点。
        if(pNode->right!=NULL){
            pNode=pNode->right;
            //当左子树不为空时一直往下遍历
            while(pNode->left!=NULL){
                pNode=pNode->left;
            }
            return pNode;
        }
//2.pNode节点没有右孩子时，pNode节点是其父结点的左孩子,直接将父节点返回.
        if(pNode->next==NULL){
            return NULL;
        }
        if(pNode==pNode->next->left){
            return pNode->next;//返回父节点
        }
//3.pNode节点没有右孩子时，pNode节点是其父结点的右孩子,需要不断的向上移动，直到父节点是其父节点的父节点的左孩子
//既然他是父节点的左节点，此时将这个节点父节点返回即可，或者遍历到了根节点，返回null.
        TreeLinkNode* pNext = pNode->next;
        while(pNext->next!=NULL){
            //父节点不为空时
            if(pNext==pNext->next->left)
            {//找到一个父节点的左子节点的节点，返回这个节点的父节点
                return pNext->next;
            }
        //继续向上找父节点
            pNext = pNext->next;
        }
        //走到这，父节点为空
        return NULL;
    }
};