ACwing 19：二叉树的下一个节点

ACwing 19：二叉树的下一个节点

思路：

• 如果当前节点的右孩子节点不为空，那么找到当前节点的右孩子的最后一个不为空的左孩子节点，最后一个不为空的左孩子节点即为中序遍历序列的下一个节点
• 如果当前节点的右孩子节点为空，那么，从父节点一直想上找，判断当前的父节点是不是其父节点的左孩子节点，如果是的话该节点的父节点就是中序遍历的下一个节点

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode *father;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL), father(NULL) {}
 * };
 */
class Solution {
public:
    TreeNode* inorderSuccessor(TreeNode* p) {
        
        // 如果当前节点有右孩子，那么右子树的最左侧节点就是当前节点的后继节点
        // 如果当前节点没有右孩子节点，那么就需要沿着父节点向上寻找，找到第一个是其父节点左孩子节点的节点
        // 那么该节点的父节点就是当前节点的后继节点。
        
        if(p->right != nullptr)
        {
            p = p->right;
            // 找到最左子节点
            while(p->left != nullptr)
            {
                 p = p->left;
            }
            return p;
        }
        
        // 如果节点的右孩子不存在 那么如果该节点是其父节点的左孩子，则返回父节点
        // 那如果该节点并不是父节点的左孩子节点，那么就继续向上遍历，查找父节点的父节点 判断条件同上。
        while(p->father != nullptr && p == p->father->right)
        {
            p = p->father;// 向上寻找父节点
            
            // p = p->father->left 停止
        }
        return p->father;// 父节点就是下一个节点
    }
};