57、剑指offer--二叉树的下一结点

题目描述

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

 

解题思路：两种情况，若果一个结点有右子树，那么其下一个结点为它的右子树中的最左节点；

若一个结点无右子树，两种情况：1）如果结点是其父结点的左子结点，那么它的下一结点为其父节点；2）如果结点是其父结点的右子节点，则沿着父节点向上遍历直到，找到一个是它父结点的左子结点的结点，如果找到，这个结点的父节点即为下一结点

/*
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;
        TreeLinkNode *pNext = NULL;
        //有右结点，其下一结点为右结点的最左子结点
        if(pNode->right != NULL)
        {
            TreeLinkNode *pRight = pNode->right;
            while(pRight->left != NULL)
                pRight = pRight->left;
            pNext = pRight;
        }
        //无右结点
        else
        {
            TreeLinkNode *pCurrent = pNode;
            TreeLinkNode *pParent = pNode->next;
            while(pParent != NULL && pCurrent == pParent->right)//是父结点的右子结点，沿父结点一直向上查找
            {
                pCurrent =pParent;
                pParent = pParent->next;
            }
            pNext = pParent;
        }
        return pNext;
    }
};