文章目录
- 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;}
};
44 ms 22.9 MB
我的CSDN博客地址 https://michael.blog.csdn.net/
长按或扫码关注我的公众号(Michael阿明),一起加油、一起学习进步!