235. 二叉搜索树的最近公共祖先
参考文章:代码随想录
参考视频:二叉搜索树找祖先就有点不一样了!| 235. 二叉搜索树的最近公共祖先_哔哩哔哩_bilibili
解题思路:
与普通二叉树求最近公共祖先方法类似,不过可以利用二叉搜索树的特性队算法进行剪枝操作,当节点大于两个值p和q时,返回左子树的结果,当小于两个值时返回右子树的结果。
public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
if (root == null || root == p || root == q) return root;
if (root.val > p.val && root.val > q.val) {
return lowestCommonAncestor(root.left, p, q);
} else if (root.val < p.val && root.val < q.val) {
return lowestCommonAncestor(root.right, p, q);
} else{
return root;
}
}
701.二叉搜索树中的插入操作
参考文章:代码随想录
参考视频:原来这么简单? | LeetCode:701.二叉搜索树中的插入操作_哔哩哔哩_bilibili
解题思路:比较简单的一道题,因为对二叉搜索树进行插入仅需要找到对应的叶子节点进行处理即可。
public TreeNode insertIntoBST(TreeNode root, int val) {
if (root == null) return new TreeNode(val);
if (root.val > val) {
root.left = insertIntoBST(root.left, val);
} else if (root.val < val) {
root.right = insertIntoBST(root.right, val);
}
return root;
}
450.删除二叉搜索树中的节点
参考文章:代码随想录
参考视频:调整二叉树的结构最难!| LeetCode:450.删除二叉搜索树中的节点_哔哩哔哩_bilibili
解题思路:要对删除节点的情况进行判断。总共归列为5种情况。
1、节点为空,直接返回null。
2、节点为叶子节点,直接返回null。
3、节点左孩子为空,右孩子不为空,返回右孩子。
4、节点右孩子为空,左孩子不为空,返回左孩子。
5、节点左右孩子都不为空。
第五种情况比较复杂,需要改变二叉搜索树的结构,这里有两种处理方法,一种是找到该节点右子树最小的值,然后将该最小节点的左子树指向原节点的左子树,最后返回原节点的右子树即可。
另一种方法也是同理,找到左子树最大的值,将该最大节点的右子树指向原节点的右子树,最后返回原节点的左子树。
这里代码列举第一种处理方法:
public TreeNode deleteNode(TreeNode root, int key) {
if (root == null) return root;
if (root.val == key) {
if (root.left == null && root.right == null) {
return null;
}
if (root.left == null && root.right != null) {
return root.right;
}
if (root.left != null && root.right == null) {
return root.left;
}
if (root.left != null && root.right != null) {
TreeNode cur = root.right;
while (cur.left != null) cur = cur.left;
cur.left = root.left;
return root.right;
}
}
if (root.val > key) {
root.left = deleteNode(root.left, key);
}
if (root.val < key) {
root.right = deleteNode(root.right, key);
}
return root;
}