代码随想录刷题day20丨235. 二叉搜索树的最近公共祖先,701.二叉搜索树中的插入操作,450.删除二叉搜索树中的节点
1.题目
1.1二叉搜索树的最近公共祖先
-
文档讲解:https://programmercarl.com/0235.%E4%BA%8C%E5%8F%89%E6%90%9C%E7%B4%A2%E6%A0%91%E7%9A%84%E6%9C%80%E8%BF%91%E5%85%AC%E5%85%B1%E7%A5%96%E5%85%88.html
-
解题思路:递归法或者迭代法
- 本题就不涉及到 前中后序了(这里没有中节点的处理逻辑,遍历顺序无所谓了)。
-
代码:
//递归法 /** * Definition for a binary tree node. * public class TreeNode { * int val; * TreeNode left; * TreeNode right; * TreeNode(int x) { val = x; } * } */ class Solution { public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) { return traversal(root,p,q); } public TreeNode traversal(TreeNode cur,TreeNode p,TreeNode q){ if(cur == null){ return null; } if(cur.val > p.val && cur.val > q.val){ TreeNode left = traversal(cur.left,p,q); if(left != null){ return left; } } if(cur.val < p.val && cur.val < q.val){ TreeNode right = traversal(cur.right,p,q); if(right != null){ return right; } } return cur; } }
//迭代法 class Solution { public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) { while (true) { if (root.val > p.val && root.val > q.val) { root = root.left; } else if (root.val < p.val && root.val < q.val) { root = root.right; } else { break; } } return root; } }
-
总结:
- 不用使用回溯,二叉搜索树自带方向性,可以方便的从上向下查找目标区间,遇到目标区间内的节点,直接返回。
1.2二叉搜索树中的插入操作
-
文档讲解:https://programmercarl.com/0701.%E4%BA%8C%E5%8F%89%E6%90%9C%E7%B4%A2%E6%A0%91%E4%B8%AD%E7%9A%84%E6%8F%92%E5%85%A5%E6%93%8D%E4%BD%9C.html
-
解题思路:递归法或者迭代法
- 这道题目其实是一道简单题目,但是题目中的提示:有多种有效的插入方式,还可以重构二叉搜索树,一下子吓退了不少人,瞬间感觉题目复杂了很多。
- 其实可以不考虑题目中提示所说的改变树的结构的插入方式。
- 只要按照二叉搜索树的规则去遍历,遇到空节点就插入节点就可以了。
-
代码:
//递归法 /** * Definition for a binary tree node. * public class TreeNode { * int val; * TreeNode left; * TreeNode right; * TreeNode() {} * TreeNode(int val) { this.val = val; } * TreeNode(int val, TreeNode left, TreeNode right) { * this.val = val; * this.left = left; * this.right = right; * } * } */ class Solution { public TreeNode insertIntoBST(TreeNode root, int val) { if(root == null){ // 如果当前节点为空,也就意味着val找到了合适的位置,此时创建节点直接返回。 TreeNode node = new TreeNode(val); return node; } if(val < root.val){ root.left = insertIntoBST(root.left,val); // 递归创建左子树 } if(val > root.val){ root.right = insertIntoBST(root.right,val); // 递归创建右子树 } return root; } }
//迭代法 class Solution { public TreeNode insertIntoBST(TreeNode root, int val) { if (root == null) return new TreeNode(val); TreeNode newRoot = root; TreeNode pre = root; while (root != null) { pre = root; if (root.val > val) { root = root.left; } else if (root.val < val) { root = root.right; } } if (pre.val > val) { pre.left = new TreeNode(val); } else { pre.right = new TreeNode(val); } return newRoot; } }
-
总结:
- 如何通过递归函数返回值完成新加入节点的父子关系赋值操作?
- 下一层将加入节点返回,本层用root->left或者root->right将其接住。
- 如何通过递归函数返回值完成新加入节点的父子关系赋值操作?
1.3删除二叉搜索树中的节点
-
文档讲解:https://programmercarl.com/0450.%E5%88%A0%E9%99%A4%E4%BA%8C%E5%8F%89%E6%90%9C%E7%B4%A2%E6%A0%91%E4%B8%AD%E7%9A%84%E8%8A%82%E7%82%B9.html
-
解题思路:递归
-
有以下五种情况:
-
第一种情况:没找到删除的节点,遍历到空节点直接返回了
-
找到删除的节点
-
第二种情况:左右孩子都为空(叶子节点),直接删除节点, 返回NULL为根节点
-
第三种情况:删除节点的左孩子为空,右孩子不为空,删除节点,右孩子补位,返回右孩子为根节点
-
第四种情况:删除节点的右孩子为空,左孩子不为空,删除节点,左孩子补位,返回左孩子为根节点
-
第五种情况:左右孩子节点都不为空,则将删除节点的左子树头结点(左孩子)放到删除节点的右子树的最左面节点的左孩子上,返回删除节点右孩子为新的根节点。
-
-
-
-
代码:
/** * Definition for a binary tree node. * public class TreeNode { * int val; * TreeNode left; * TreeNode right; * TreeNode() {} * TreeNode(int val) { this.val = val; } * TreeNode(int val, TreeNode left, TreeNode right) { * this.val = val; * this.left = left; * this.right = right; * } * } */ 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.left; }else if(root.left == null && root.right != null){ return root.right; }else{ TreeNode cur = root.right; while(cur.left != null){ cur = cur.left; } cur.left = root.left; return root.right; } } if(key < root.val){ root.left = deleteNode(root.left,key); } if(key > root.val){ root.right = deleteNode(root.right,key); } return root; } }
-
总结:
- 搜索树的节点删除要比节点增加复杂的多,有很多情况需要考虑