剑指offer面试题8：二叉树的下一个节点

/*
 题目：给定一个二叉树的中序遍历和一个节点，找出这个节点在中序遍历中的下一个节点，
 树中的节点除了有左右指针还有一个指向父节点的指针
         			a
                /      \
             b            c
           /   \        /   \
        d       e      f     g
               /  \
             h     i
             它的中序遍历是{d,b,h,e,i,a,f,c,g}
 */
/*
 1.如果一个节点有右子树，那么它的下一个节点就是它的右子树中最左子节点，例如b的下一个是h
 2.如果一个节点没有右子树，如果这个节点是它父节点的左节点，那么它的下一个节点就是它的父节点
 3.如果一个节点既没有右子树，这个节点还是它父节点的右子节点，那么我们可以沿着父节点的指针一直向上遍历，直到找到一个是它父节点的左子节点的节点，那么这个节点的父节点就是我们要找的下一个节点
 */

BinaryTreeNode *GetNext(BinaryTreeNode *pNode)
{
    if(pNode==nullptr)
        return nullptr;
    
    BinaryTreeNode*pNext=nullptr;
    if(pNode->right!=nullptr)
    {
        BinaryTreeNode *pRight=pNode->right;
        while(pRight->left !=nullptr)
            pRight=pRight->left;
        pNext=pRight;
    }
    else if(pNode->parent != nullptr)
    {
        BinaryTreeNode *pCur=pNode;
        BinaryTreeNode *pparent=pNode->parent;
        while(pparent!=nullptr && pCur==pparent->right)
        {
            pCur=pparent;
            pparent=pparent->parent;
        }
        pNext=pparent;
    }
    return pNext;
}