import com.sun.source.doctree.EscapeTree; import java.security.spec.RSAOtherPrimeInfo; import java.util.Map; public class Day17 { private static class TreeNode { int val; TreeNode left; TreeNode right; public TreeNode(int val) { this.val = val; } public TreeNode(int val, TreeNode left, TreeNode right) { this.val = val; this.left = left; this.right = right; } } //给定一个二叉搜索树, 找到该树中两个指定节点的最近公共祖先。 //百度百科中最近公共祖先的定义为:“对于有根树 T 的两个结点 p、q,最近公共祖先表示为一个结点 x, // 满足 x 是 p、q 的祖先且 x 的深度尽可能大(一个节点也可以是它自己的祖先)。” public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) { //本题就是标准的搜索一条边的写法,遇到递归函数的返回值,如果不为空,立刻返回。 //二叉搜索树可以依靠val的大小判断递归哪条边 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; } //给定二叉搜索树(BST)的根节点 root 和要插入树中的值 value ,将值插入二叉搜索树。 返回插入后二叉搜索树的根节点。 // 输入数据 保证 ,新值和原始二叉搜索树中的任意节点值都不同。 //注意,可能存在多种有效的插入方式,只要树在插入后仍保持为二叉搜索树即可。 你可以返回 任意有效的结果 。 public TreeNode insertIntoBST(TreeNode root, int val) { if(root==null) return new TreeNode(val); if(val<root.val) root.left= insertIntoBST(root.left,val); else root.right= insertIntoBST(root.right,val); return root; } //迭代法 public TreeNode insertIntoBST2(TreeNode root, int val) { if (root==null) return new TreeNode(val); TreeNode newRoot=root; while (root!=null){ if(val>root.val){ if(root.right==null) break; root=root.right; } else { if(root.left==null) break; root=root.left; } } if(val<root.val) root.left= new TreeNode(val); else root.right= new TreeNode(val); return newRoot; } //给定一个二叉搜索树的根节点 root 和一个值 key,删除二叉搜索树中的 key 对应的节点,并保证二叉搜索树的性质不变。 // 返回二叉搜索树(有可能被更新)的根节点的引用。 //一般来说,删除节点可分为两个步骤: //首先找到需要删除的节点; //如果找到了,删除它。 //二叉搜索树删除节点会改变树的结构,将删除情况分开,分别处理 public TreeNode deleteNode(TreeNode root, int key) { if(root==null) return null;//没找到 if(key==root.val){ //分四种情况 if(root.left==null&&root.right==null) return null;//叶子节点 else if(root.right!=null&&root.left==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 (key<root.val) root.left=deleteNode(root.left,key); if(key>root.val) { root.right=deleteNode(root.right,key); } return root; } }
代码随想录|235. 二叉搜索树的最近公共祖先 701.二叉搜索树中的插入操作 450.删除二叉搜索树中的节点
最新推荐文章于 2024-08-20 15:39:11 发布