LeetCode 285. 二叉搜索树中的顺序后继（中序遍历）

文章目录

- 1. 题目
- 2. 解题

1. 题目

给你一个二叉搜索树和其中的某一个结点，请你找出该结点在树中顺序后继的节点。

结点 p 的后继是值比 p.val 大的结点中键值最小的结点。

示例 1:
在这里插入图片描述

输入: root = [2,1,3], p = 1
输出: 2
解析: 这里 1 的顺序后继是 2。
请注意 p 和返回值都应是 TreeNode 类型。

示例 2:
在这里插入图片描述

输入: root = [5,3,6,2,4,null,null,1], p = 6
输出: null
解析: 因为给出的结点没有顺序后继，所以答案就返回 null 了。注意:
假如给出的结点在该树中没有顺序后继的话，请返回 null
我们保证树中每个结点的值是唯一的

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/inorder-successor-in-bst
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

2. 解题

类似题目：LeetCode 510. 二叉搜索树中的中序后继 II（查找右子树或者祖父节点）

循环版中序遍历，找到p节点后的下一个即是答案

class Solution {
public:TreeNode* inorderSuccessor(TreeNode* root, TreeNode* p) {if(!root) return NULL;stack<TreeNode*> stk;TreeNode *cur;bool found = false;while(root || !stk.empty()){while(root){stk.push(root);root = root->left;}cur = stk.top();if(found)return cur;stk.pop();root = cur->right;if(cur == p)found = true;}return NULL;}
};