代码随想录算法营DAY22 | 235. 二叉搜索树的最近公共祖先 701.二叉搜索树中的插入操作 450.删除二叉搜索树中的节点
235. 二叉搜索树的最近公共祖先
题目链接: 235. 二叉搜索树的最近公共祖先
思路:
这一题二叉搜索树比二叉树简单一些,从上到下遍历,最先出现在p,q区间内的数就是最近公共祖先。
用递归也可以,用迭代也可以。
代码:
//递归
var lowestCommonAncestor = function(root, p, q) {
if(root.val>p.val&&root.val>q.val) {
let left = lowestCommonAncestor(root.left,p,q);
if(left) return left;
}
else if(root.val<p.val&&root.val<q.val) {
let right = lowestCommonAncestor(root.right,p,q);
if(right) return right;
}
else return root;
};
//迭代
var lowestCommonAncestor = function(root, p, q) {
// 使用迭代的方法
while(root) {
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 {
return root;
}
}
return null;
};
701.二叉搜索树中的插入操作
题目链接: 701.二叉搜索树中的插入操作
思路:
这道题目其实是一道简单题目,但是题目中的提示:有多种有效的插入方式,还可以重构二叉搜索树,一下子吓退了不少人,瞬间感觉题目复杂了很多。
其实可以不考虑题目中提示所说的改变树的结构的插入方式。
只要遍历二叉搜索树,找到空节点 插入元素就可以了,那么这道题其实就简单了。
代码:
var insertIntoBST = function (root, val) {
if(root===null) return root=new TreeNode(val);
if (val > root.val) {
if (!root.right) {
root.right = new TreeNode(val);
return root;
}
else root.right = insertIntoBST(root.right, val);
}
if (val < root.val) {
if (!root.left) {
root.left = new TreeNode(val);
return root;
}
else root.left = insertIntoBST(root.left, val);
}
return root;
};
//优化版本
var insertIntoBST = function (root, val) {
if(root===null) return root=new TreeNode(val);
if (val > root.val) root.right = insertIntoBST(root.right, val);
if (val < root.val)root.left = insertIntoBST(root.left, val);
return root;
};
450.删除二叉搜索树中的节点
题目链接: 450.删除二叉搜索树中的节点
思路:
这道题和上一题就不同,涉及到树的结构调整。
这里就把二叉搜索树中删除节点遇到的情况都搞清楚。
有以下五种情况:
第一种情况:没找到删除的节点,遍历到空节点直接返回了
找到删除的节点
第二种情况:左右孩子都为空(叶子节点),直接删除节点, 返回NULL为根节点
第三种情况:删除节点的左孩子为空,右孩子不为空,删除节点,右孩子补位,返回右孩子为根节点
第四种情况:删除节点的右孩子为空,左孩子不为空,删除节点,左孩子补位,返回左孩子为根节点
第五种情况:左右孩子节点都不为空,则将删除节点的左子树头结点(左孩子)放到删除节点的右子树的最左面节点的左孩子上,返回删除节点右孩子为新的根节点。
代码:
var deleteNode = function(root, key) {
if(root===null) return null;
if(root.val<key) root.right=deleteNode(root.right,key);
else if(root.val>key) root.left=deleteNode(root.left,key);
else{
if(!root.left&&!root.right) return null;
else if(!root.right) return root.left;
else if(!root.left) return root.right;
else {
let node=root.right;
while(node.left) node=node.left;
node.left=root.left;
return root.right;
}
}
return root;
};
感想总结
虽然是二刷,但是基本忘记了怎么做,都是先看一下思路然后手写代码,通过倒是没什么问题。