【剑指Offer】二叉树的下一个结点

1.题目

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

2.Code

中序遍历：左子树、根、右子树

中序遍历的结果：4,2,8,5,9,1,6,3,7

观察图片可以知道：节点2的下一节点是8，即右子树的最左节点

节点9的下一节点是1，这个就比较麻烦了，它没有右子树，则只能通过指向父结点的指针回溯，同时判断是不是右孩子即可

测试结果

/*
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* pRes = NULL;
        TreeLinkNode* pTmp = NULL;
        TreeLinkNode* pCur = NULL;

        if(pNode->right){
            pTmp = pNode->right;
            while(pTmp->left) {
                pTmp = pTmp->left;
            }
            pRes = pTmp;
        }
        else if(pNode->next){
            pTmp = pNode;
            pCur = pNode->next;
            while(pCur && pTmp == pCur->right){
                pTmp = pCur;
                pCur = pCur->next;
            }
            pRes = pCur;
        }
        return pRes;
    }
};