原题目
面试题 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
第一遍解法
中序遍历二叉搜索树,元素值递增。用pre来保存上一个节点,若上一个节点与p相等(pre == p),则返回当前节点(root)。
class Solution {
private TreeNode pre = null;
public TreeNode inorderSuccessor(TreeNode root, TreeNode p) {
if (root == null) {
return null;
}
TreeNode L = inorderSuccessor(root.left, p);
if (pre != null && pre == p) {
pre = root; // 这一步是必要的,不然后续pre将一直等于p,最终返回根节点。
return root;
}
pre = root;
TreeNode R = inorderSuccessor(root.right, p);
if (L != null) {
return L;
}
if (R != null) {
return R;
}
return null;
}
}
时间复杂度: O(n)
空间复杂度: O(n)
网上好的解法
我自己太死板了,遇到线索二叉树就一直想着中序遍历,当然这没有问题,但是在空间复杂度上会比较高。
这道题目告诉自己,就算遇到线索二叉树也要具体分析。
因为这题就是想找一个与p一样的节点,根据线索二叉树的性质,若p小于root就在左子树去找,若p大于root就去右子树找,依次往下找就可以了。
class Solution {
public TreeNode inorderSuccessor(TreeNode root, TreeNode p) {
//问题的本质是找到最靠近p节点且值大于p节点值的那个节点
TreeNode res = root;
//设定临时变量方便对树的操作
TreeNode temp = root;
while (temp != null){
//如果当前节点的值小于等于目标节点的值,那一定不是答案节点,且答案节点在该节点的右孩子中
if (temp.val <= p.val) {
temp = temp.right;
} else {
//如果当前节点的值大于目标节点的值,那么该节点有可能是答案节点,具体是不是需要遍历其左孩子,寻找更靠近p节点值的答案
res = temp;
temp = temp.left;
}
}
return res.val <= p.val ? null : res;
}
}
作者:ShiningShar
链接:https://leetcode-cn.com/problems/successor-lcci/solution/javashuang-bai-by-shiningshar-2/
来源:力扣(LeetCode)
时间复杂度: O(n)
空间复杂度: O(1)
最后的代码
与网上代码一致。
class Solution {
public TreeNode inorderSuccessor(TreeNode root, TreeNode p) {
TreeNode res = root;
TreeNode temp = root;
while (temp != null){
if (p.val >= root.val) {
temp = temp.right;
} else {
res = temp;
temp = temp.left;
}
}
return res.val <= p.val ? null : res;
}
}
时间复杂度: O(n)
空间复杂度: O(1)
小结
- 无