代码随想录–二叉树部分
day 20 二叉树第七天
一、力扣235–二叉搜索树的最近公共祖先
代码随想录题目链接:代码随想录
给定一个二叉搜索树, 找到该树中两个指定节点的最近公共祖先。
百度百科中最近公共祖先的定义为:“对于有根树 T 的两个结点 p、q,最近公共祖先表示为一个结点 x,满足 x 是 p、q 的祖先且 x 的深度尽可能大(一个节点也可以是它自己的祖先)。
实际上用236的做法可以直接解答,不过考虑到这是有序的二叉搜索树,没必要那么麻烦
考虑到其为有序的,所以中序遍历的话是递增的,也就是说最近公共祖先一定是处于两个数之间的
而最近怎么体现?自上向下遍历,第一个处于两个数之间的节点就是最近公共祖先了
如果当前节点比两个数都小,那就向其右边继续向下,这样才能让数字增加,反之亦然
代码如下:
class Solution {
public:
TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
if(root->val < p->val && root->val < q->val) return lowestCommonAncestor(root->right, p, q);
else if(root->val > p->val && root->val > q->val) return lowestCommonAncestor(root->left, p , q);
else return root;
}
};
二、力扣701–二叉树种的插入操作
代码随想录题目链接:代码随想录
给定二叉搜索树(BST)的根节点 root 和要插入树中的值 value ,将值插入二叉搜索树。 返回插入后二叉搜索树的根节点。 输入数据 保证 ,新值和原始二叉搜索树中的任意节点值都不同。
最简单的做法是中序遍历,然后把数字插进去再重新构建二叉搜索树,慢但是有效
但是没必要,只需要比较每个节点和要插入数的大小即可,如果当前节点更大,那就向左遍历,反之亦然
一直到当前节点为空,说明到了尾巴,在这里插入新节点就行
代码如下:
class Solution {
public:
TreeNode* insertIntoBST(TreeNode* root, int val) {
if(!root) return new TreeNode(val);
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 对应的节点,并保证二叉搜索树的性质不变。返回二叉搜索树(有可能被更新)的根节点的引用。
一般来说,删除节点可分为两个步骤:
首先找到需要删除的节点;
如果找到了,删除它。
要考虑多种情况:没找到、找到了且是叶子节点、找到了且只有左/右子树、找到了且同时有左右子树
前四种都好处理,返回空或者删除当前节点,用子树补位
最后一种是最麻烦的,删除了节点还要安排左右子树的连接
比较方便的做法是左子树直接接到右子树的最左叶子上,然后用这个新的子树去补位
代码如下:
class Solution {
public:
TreeNode* deleteNode(TreeNode* root, int key) {
if(!root) return root;
if(root->val == key)
{
if(!root->left && !root->right)
{
delete root;
return nullptr;
}
else if(!root->left)
{
auto temp = root->right;
delete root;
return temp;
}
else if(!root->right)
{
auto temp = root->left;
delete root;
return temp;
}
else
{
TreeNode * leaf = root->right;
while(leaf->left)leaf = leaf->left;
leaf->left = root->left;
auto temp = root->right;
delete root;
return temp;
}
}
if(root->val > key) root->left = deleteNode(root->left, key);
if(root->val < key) root->right = deleteNode(root->right, key);
return root;
}
};