剑指Offer编程题--二叉树的下一个结点

题目链接：二叉树的下一个结点

题意：

   给一个二叉树中的一个结点，输出此结点中序遍历情况下的下一个结点

解题思路；

   在二叉树中，中序遍历首先遍历左子树，然后访问根结点，最后遍历右子树。

 三种情况：

   1. 如果当前结点有右子树，那么下一个结点就是右子树的最左节点

  2. 如果没有右子树

           （1）如果该节点为 该父节点的左孩子，那么下一个结点就是 该父节点

          （2）往上遍历，知道满足上面的情况

3. 前两种都不符合，说明此节点为最后一个点，返回NULL

/*
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;
        }
        if(pNode->right != NULL){ // 有右子树
            pNode = pNode->right;
            while(pNode->left != NULL){ // 查找最左节点
                pNode = pNode->left;
            }
            return pNode;
        }
        else{
            while(pNode->next != NULL){
                if(pNode->next->left == pNode){ // 该节点为它父节点的左孩子
                    return pNode->next;
                }
                pNode = pNode->next;  // 向上遍历
            }
            return NULL;  // 该节点为最后一个点
        }
    }
};