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

1.题目描述

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

2.问题分析

3.源代码

// 例如：  1
//     2     3
//  4      6   7
// 中序遍历是 4 2 1 6 3 7

我们分成两种情况：

• 一种是:该节点的右子树不为空，查找该右子树的中序遍历的第一个节点，即：最左节点。例如：pNode是1，那么下一个节点就是6；
• 另外一种是:该节点的右子树为空，那么需要回溯到该节点的父节点，来查找。如果pNode是2，那么下一个节点应该是1，我们发现1的左节点就是2节点，所以下一个节点是1。

TreeLinkNode* GetNext(TreeLinkNode* pNode)
{
    if(pNode == NULL)
        return NULL;
    //右子树存在，找右子树的中序遍历第一个节点,即：第一个左节点为空的节点
    if(pNode->right != NULL)
    {
        TreeLinkNode* pRoot = pNode->right;
        while(pRoot->left != NULL)
            pRoot = pRoot->left;
        return pRoot;
    }
    else
    {
        //找到第一个父节点的左节点与该节点相等，那么该父节点就是我们要找的下一个节点
        // 例如：  1
        //     2     3
        //  4      6   7
        // 中序遍历是 4 2 1 6 3 7
        //如果pNode是2，那么下一个节点应该是1，我们发现1的左节点就是2节点，所以下一个节点是1
        while(pNode->next != NULL)
        {
            TreeLinkNode* parent = pNode->next;
            if(parent->left == pNode)
                return parent;
            pNode = parent;
        }
        return NULL;
    }
}