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

题目描述
给定一个二叉树和其中的一个结点，请找出中序遍历顺序的下一个结点并且返回。注意，树中的结点不仅包含左右子结点，同时包含指向父结点的指针。
思路：
如果这个节点为空，返回NULL
若果这个节点刚好是一个孤立的根节点，返回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->left==NULL&&pNode->right==NULL&&pNode->next==NULL)//这个节点是个孤立的根节点
        {
            return NULL;
        }
        if(pNode->left==NULL&&pNode->right==NULL&&pNode->next->left==pNode)//左叶节点
        {
            return pNode->next;//左叶节点的父节点
        }
        if(pNode->left==NULL&&pNode->right==NULL&&pNode->next->right==pNode)//判断是不是最右的右叶节点
        {
            TreeLinkNode *q = pNode;
            while(q->next!=NULL)//一直寻找到根节点
            {
                q = q->next;
            }
            while(q->right!=NULL)//又遍历回最右的叶节点
            {
                q = q->right;
            }
            if(pNode==q)
            {
                return NULL;
            }
            return pNode->next->next;//右叶节点的父节点的父节点
        }
       
        TreeLinkNode *p = pNode;//不是叶节点那就是中间的一个节点了，中间任意一个节点
        if(p->right!=NULL)
        {
             p=p->right;
            while(p->left!=NULL)
            {
                p=p->left;
            }
            return p;
        }
        
        
        return p->next;
        
    }
};

finished!!!