给你一个二叉搜索树和其中的某一个结点,请你找出该结点在树中顺序后继的节点。
结点 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 著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
类似题目: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
我的博客地址 https://michael.blog.csdn.net/
长按或扫码关注我的公众号(Michael阿明),一起加油、一起学习进步!
