【二叉树】二叉树的下一个节点

问题描述

给定一棵二叉树和其中的一个节点，如何找出中序遍历顺序的下一个节点。
树中的节点除了有两个左右指针，还有一个指向父节点的指针。

分析问题

可以分为三种情况：
(1)节点有右子树，找右子树中最左边的节点
例如图中结点b的下一个结点是h，结点a的下一结点是f。

(2)节点没有右子树，且是其父节点的左子树，下个节点就是父节点
图中结点d的下一个结点是b，f的下一个结点是c。

(3)节点没有右子树，且是父节点的右子树，找下个节点，一直向上遍历，直到遇到一个节点是其父节点的左子树，那么父节点就是下个节点;如果没有遍历到根节点的话，说明没有。

例如，对于结点i，我们首先找到其父亲结点e，e是其父亲结点的右子结点，所以我们继续找到b，b是其父亲结点的左子结点。所以i的下一个结点就是b的父亲结点，即a。对于结点g，我们首先沿着指向父亲结点的指针找到c，c是其父亲结点的右子结点，所以我们继续找到a。因为a是树的根节点，没有父亲结点，所以g的下一个结点不存在。

Node* GetNext(Node* pNode)
{
    if (pNode == NULL)
        return NULL;

    Node* pNext = NULL;
    //（1）
    if (pNode->right != NULL)
    {
        //找右子树中最左边的节点
        pNext = pNode->right;
        while (pNext->left)
            pNext = pNext->left;        
    }
    //（2）（3）
    else if (pNode->parent != NULL)
    {
        Node* pCur = pNode;
        Node* pParent = pNode->parent;
        //（3）,如果没有执行(3)，结果就是2
        while (pParent != NULL && pCur == pParent->right)
        {
            pCur = pParent;
            pParent = pParent->parent;
        }
        pNext = pParent;
    }
    return pNext;
}