剑指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;
     }
 };