【leetcode】510. 二叉搜索树中的中序后继 II( Inorder Successor in BST II )

---

题目描述

【leetcode】510. 二叉搜索树中的中序后继 II( Inorder Successor in BST II )

给定一棵二叉搜索树和其中的一个节点，找到该节点在树中的中序后继。
一个结点 p 的中序后继是键值比 p.val大所有的结点中键值最小的那个。
你可以直接访问结点，但无法直接访问树。每个节点都会有其父节点的引用。

示例 1:

输入: [2,1,3]
输出: 2
解析: 1的中序后继结点是2。注意p和返回值都是Node类型的。

示例 2:

输入:
输出: null
解析: 该结点没有中序后继，因此返回null。

示例 3:

输入:
输出: 17

示例 4:

输入:
输出: 15

注意:

• 如果给定结点不存在中序后继，返回null。
• 树中各结点的值均保证唯一。
• 注意我们使用的是Node类型而不是TreeNode类型，它们的字符串表示不一样。

第一次解答

思路
错了很多次才完善思路。

1. 若node有右孩子，则返回右子树中序遍历的第一个结点（即右子树最左边的结点）
2. 否则若node无右孩子，查看自己的父结点
   2.1 若父结点为空，返回nullptr
   2.2 若自己是父结点的左孩子，则返回父结点
   2.3 若自己是父结点的右孩子，继续查看父亲的父亲

简单来说就是，当node有右孩子时，node后继结点为右子树的最小结点，即右子树中最左下角的结点；当node结点无右孩子时，node后继结点要么为空，要么为其父结点或祖先，并且要要求node在这个父亲或祖先的左子树上。

test case:
[2,1,3]
1
[15,6,18,3,7,17,20,2,4,null,13,null,null,null,null,null,null,null,null,9]
15
[15,6,18,3,7,17,20,2,4,null,13,null,null,null,null,null,null,null,null,9]
13

代码：

/*
// Definition for a Node.
class Node {
public:
    int val;
    Node* left;
    Node* right;
    Node* parent;
};
*/
class Solution {
public:
    Node* inorderSuccessor(Node* node) {
        if(nullptr == node)
            return nullptr;
        if(nullptr != node->right){
            // 对右子树中序遍历，返回第一个元素
            node = node->right;
            while(nullptr != node->left){
                node = node->left;
            }
            return node;
        }
        while(nullptr != node->parent && node->parent->left != node){
            node = node->parent;
        }
        return node->parent;

    }
};

结果：

相关/参考链接