面试题 04.06. 后继者(2022-5-16)
设计一个算法,找出二叉搜索树中指定节点的“下一个”节点(也即中序后继)。
如果指定节点没有对应的“下一个”节点,则返回null。
示例 1:
输入: root = [2,1,3], p = 1
2
/ \
1 3
输出: 2
示例 2:
输入: root = [5,3,6,2,4,null,null,1], p = 6
5
/ \
3 6
/ \
2 4
/
1
输出: null
解题思路
每次遇到这种题,就楞半天,还是写的太少,整体思路也不难,只是差点细节和经验,需要参考别人的才能茅塞顿开。
整体思路:二叉搜索树的下一个节点,也就是次大的节点,如果有右子树,那就是是右子树的最左叶节点,没有就是它的父节点。
首先尝试 p
节点有没有右子树,有了直接遍历返回,没有了再去找他的位置(找的同时,保存父节点)找到了返回父节点就好了
var inorderSuccessor = function(root, p) {
let explored = null,node = root
if(p.right){
p = p.right
while(p.left){
p = p.left
}
return p
}
while(node != null){
if(node.val > p.val){
[explored, node] = [node, node.left]
}else if(node.val < p.val){
node = node.right
}else{
return explored
}
}
return null
};
来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/successor-lcci
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。