算法：在二叉树中找到一个节点的后继节点

题目描述

在二叉树的中序遍历的序列中，node的下一个节点叫作node的后继节点。

现在有一种新的二叉树节点类型如下：

struct TreeNode {
    int val;
    TreeNode *left;
    TreeNode *right;
    TreeNode *parent;
  
    TreeNode(int x) : val(x), left(nullptr), right(nullptr), parent(nullptr) {}

};

该结构比普通二叉树节点结构多了一个指向父节点的parent指针。 假设有一棵Node类型的节点组成的二叉树，树中每个节点的parent指针都正确地指向自己的父节点，头节点的parent指向null。只给一个在二叉树中的某个节点 node，请实现返回node的后继节点的函数。在二叉树的中序遍历的序列中，node的下一个节点叫作node的后继节点。

比如下图中：

题目解析

分析：
（1）如果node有右子树，那么后继节点就是右子树上最左边的节点

（2）如果node没有右子树，那么先看node是不是node父节点的左孩子：

• 如果是左孩子，那么此时node的父节点就是node的后继节点
  
  

• 如果是右孩子，那么向上选择node的后继节点，假设向上移动到的节点记为s，s的父节点记为p，如果发现s是p的左孩子，那么节点p就是node节点的后继节点

（3）如果在情况2中一直往上寻找，都找到了NULL是也没有找到node的后继节点，说明node压根不存在后继节点

class Solution {
    TreeNode * getLeftMost(TreeNode * node){
        if(node == nullptr){
            return nullptr;
        }
        
        while (node->left != nullptr){
            node = node->left;
        }
        return node;
    }
public:
    TreeNode * getNextNode(TreeNode *node) {
        if(node == nullptr){
            return node;
        }
        
        if(node->right != nullptr){
            //如果node有右子树，那么后继节点就是右子树上最左边的节点
            return getLeftMost(node->right);
        }else{
            TreeNode *parent = node->parent;
            while (parent != nullptr && parent->left != node){
                node = parent;
                parent = node->parent;
            }
            return parent;
        }
    }
};

复杂度分析：如果node节点和node后继节点之间的实际距离为L，这里只用走过L个节点，时间复杂度为O(L)，空间复杂度为O(1)