面试题 04.06. 后继者
设计一个算法,找出二叉搜索树中指定节点的“下一个”节点(也即中序后继)。
如果指定节点没有对应的“下一个”节点,则返回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
解法一
public static TreeNode inorderSuccessor(TreeNode root, TreeNode p) {
if (root == null) {
return null;
}
// 当前节点值小于等于目标值,那么当前目标值的后继者必然在右子树
if (p.val >= root.val) {
return inorderSuccessor(root.right, p);
}
// 否则结果有可能是当前节点,或者在当前节点的左子树中
// 那么先去左子树找一下试试,找不到的话返回当前节点即是结果
TreeNode node = inorderSuccessor(root.left, p);
return node == null ? root : node;
}
解法二
public static TreeNode inorderSuccessor(TreeNode root, TreeNode p) {
int target = p.val;
TreeNode cur = root;
TreeNode ans = null;
while (cur != null) {
if (cur.val > target) {
ans = cur;
cur = cur.left;
} else {
cur = cur.right;
}
}
return ans;
}
网友评论