235. 二叉搜索树的最近公共祖先
● 力扣题目链接
● 给定一个二叉搜索树, 找到该树中两个指定节点的最近公共祖先。
思路
● 和普通二叉树相比,简单很多
● 如果落在了[p, q]之间,返回root;都小就向右看,都大就向左看
代码
class Solution {public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {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);return root;}
}class Solution {public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {while (root != null) {if (root.val < p.val && root.val < q.val) root = root.right;else if (root.val > p.val && root.val > q.val) root = root.left;else return root;}return root;}
}
701.二叉搜索树中的插入操作
● 力扣题目链接
● 给定二叉搜索树(BST)的根节点和要插入树中的值,将值插入二叉搜索树。 返回插入后二叉搜索树的根节点。 输入数据保证,新值和原始二叉搜索树中的任意节点值都不同。
● 注意,可能存在多种有效的插入方式,只要树在插入后仍保持为二叉搜索树即可。 你可以返回任意有效的结果。
思路
● 二叉搜索树,找到待插入的位置,我需要保留上一个节点,然后看是放在左叶子还是右叶子
● 递归也可以,这个函数的作用是找到合适的插入位置,并返回根节点
○ 如果root的val小,那就往右找,返回的根节点是root的右子节点
代码
class Solution {public TreeNode insertIntoBST(TreeNode root, int val) {if (root == null) return new TreeNode(val);TreeNode pre = root;TreeNode node = root;while (node != null) {pre = node;if (node.val < val) node = node.right;else node = node.left;}TreeNode insert = new TreeNode(val);if (pre.val < val) {pre.right = insert;} else {pre.left = insert;}return root;}
}class Solution {public TreeNode insertIntoBST(TreeNode root, int val) {if (root == null) return new TreeNode(val);if (root.val < val) {root.right = insertIntoBST(root.right, val);} else {root.left = insertIntoBST(root.left, val);}return root;}
}
450.删除二叉搜索树中的节点
● 力扣题目链接
● 给定一个二叉搜索树的根节点 root 和一个值 key,删除二叉搜索树中的 key 对应的节点,并保证二叉搜索树的性质不变。返回二叉搜索树(有可能被更新)的根节点的引用。
思路
● 注意讨论五种情况
代码
class Solution {public TreeNode deleteNode(TreeNode root, int key) {// 没找到if (root == null) return null;// 找到了if (root.val == key) {// 左子节点空if (root.left == null) return root.right;// 右子节点空if (root.right == null) return root.left;// 都不空TreeNode r = root.right;// 找右子树的最左侧叶子节点while (r.left != null) {r = r.left;}// 交换r.left = root.left;// 返回新rootreturn root.right;}// 暂时没找到,继续递归if (root.val < key) root.right = deleteNode(root.right, key);else if (root.val > key) root.left = deleteNode(root.left, key);return root;}
}