剑指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->right == NULL) {
            if(pNode->next == NULL) return NULL;
            TreeLinkNode* p1 = pNode;
            TreeLinkNode* p2 = pNode->next;
            while(p2 && p2->right == p1) {
                p1 = p1->next;
                p2 = p1->next;
            }
            if(p2) return p2;
            return NULL;
        }
        TreeLinkNode* nextNode = pNode->right;
        while(nextNode->left) {
            nextNode = nextNode->left;
        }
        return nextNode;
    }
};