235. 二叉搜索树的最近公共祖先
题目
给定一个二叉搜索树, 找到该树中两个指定节点的最近公共祖先。
百度百科中最近公共祖先的定义为:“对于有根树 T 的两个结点 p、q,最近公共祖先表示为一个结点 x,满足 x 是 p、q 的祖先且 x 的深度尽可能大(一个节点也可以是它自己的祖先)。”
例如,给定如下二叉搜索树: root = [6,2,8,0,4,7,9,null,null,3,5]
思路
本题递归法和迭代法都挺简单,终止条件:不能为空 递归参数:节点 p q
单层递归:如果节点的值比pq的值都大,向左遍历,都小向右遍历。
代码
class Solution {
public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
if (root == null) {
return null;
}
if (root.val > p.val && root.val > q.val && root.left != null) {
return lowestCommonAncestor(root.left, p, q);
}
if (root.val < p.val && root.val < q.val && root.right != null) {
return lowestCommonAncestor(root.right, p, q);
}
return root;
}
}
易错点
比较值的时候,左右别写反了就好。
701.二叉搜索树中的插入操作
题目
给定二叉搜索树(BST)的根节点 root
和要插入树中的值 value
,将值插入二叉搜索树。 返回插入后二叉搜索树的根节点。 输入数据 保证 ,新值和原始二叉搜索树中的任意节点值都不同。
注意,可能存在多种有效的插入方式,只要树在插入后仍保持为二叉搜索树即可。 你可以返回 任意有效的结果 。
思路
本题采取递归法,终止条件:判断是否为空,为空的话开始创建新的节点,并返回到上一层
参数:root和val
单层递归:如果节点的值大于val则向左遍历,小的话向右遍历。最后返回根节点
代码
class Solution {
public TreeNode insertIntoBST(TreeNode root, int val) {
if (root == null) {
TreeNode newNode = new TreeNode(val);
return newNode;
}
if (root.val > val) {
root.left = insertIntoBST(root.left, val);
}
if (root.val < val) {
root.right = insertIntoBST(root.right, val);
}
return root;
}
}
易错点
遇到空就创建新节点,最后返回根节点。
450.删除二叉搜索树中的节点
题目
给定一个二叉搜索树的根节点 root 和一个值 key,删除二叉搜索树中的 key 对应的节点,并保证二叉搜索树的性质不变。返回二叉搜索树(有可能被更新)的根节点的引用。
一般来说,删除节点可分为两个步骤:
- 首先找到需要删除的节点;
- 如果找到了,删除它。
思路
本题依然采取递归的方法做,在递归的终止条件中处理细节。
终止条件:
节点为空,返回空节点。节点的值等于key。
有四种情况,叶子节点,直接删除
非叶子节点,左空右不空,返回右节点
左不空右空,返回左节点
左右都不空,设置一个临时节点cur遍历到右子树的左边直到不为空。
把左子树放在cur的左边,返回右子树。
单层递归:比key大向右遍历,比key小向左遍历。最后返回根节点。
代码
class Solution {
public TreeNode deleteNode(TreeNode root, int key) {
if (root == null) {
return null;
}
if (root.val == key) {
if (root.left == null && root.right == null) {
return null;
} else if (root.left == null && root.right != null) {
return root.right;
} else if (root.left != null && root.right == null) {
return root.left;
} else {
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;
}
}
易错点
关键点在于处理非叶子节点的左右都不为空的情况,我这采取的是右子树继位,所以才向右子树的左孩子遍历。
总结
越刷思路越清晰,继续加油!
不积细流,无以成江河。