一、前言
参考文献:代码随想录
这是第八天二叉树了,感觉对自我的提升还是有很多的,但是还需多多刷题,掌握二叉树技巧。
二、搜索二叉树最进的公共祖先
1、思路:
关于最近公共祖先,我第一个想到的还是昨天二叉树的最近公共祖先,顺便复习一下。
代码如下:
class Solution {
public:
TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
if (root == NULL || root == p || root == q) return root;
// 左
TreeNode* left = lowestCommonAncestor(root->left, p, q);
// 右
TreeNode* right = lowestCommonAncestor(root->right, p ,q);
// 中
if (right == NULL && left != NULL) return left;
else if (right != NULL && left == NULL) return right;
else if (right != NULL && left != NULL) return root;
else return NULL;
}
};
但是,在这里我们可以利用二叉搜索树的性质,让代码变得更加简单,我总结了一下几点:
(1)返回值还是TreeNode类型,传入参数还是那几个;
(2)终止条件依旧为
if (root == NULL) return root;
(3)采用的是后序遍历,主要判断就是所遍历的值是否在p,q这个区间,在这个区间出现的第一个几点就为这个公共节点,理解这个就是解题的关键,在小于这个区间,就往左子树遍历,大于就往右子树遍历,这就是核心思路,这里一定要返回各个函数,因为公共祖先需要传递到根节点返回
2、代码如下:
class Solution {
private:
TreeNode* getAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
if (root == NULL) return root;
// 左
if (p->val < root->val && q->val < root->val) {
// 适当的返回,传递公共祖先
return getAncestor(root->left, p , q);
}
// 右
else if (p->val > root->val && q->val > root->val) {
return getAncestor(root->right, p ,q);
}
// 中
else {
return root;
}
}
public:
TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
return getAncestor(root, p , q);
}
};
三、二叉树中的插入操作
1、思路:
我最开始的想法是找到了最底层的第一个叶子节点,然后接入就行了,这是迭代法,这不是今天的主题,接着就是学了卡哥和GPT教授的思路,写出了一个递归法,有两个小版本,第一个比较繁琐:
(1)返回值为TreeNode,传入参数就为root和val
(2)遍历到NULL时,创建一个新节点返回给父节点
(3)在BST树中并不需要遍历全部的节点,只需要用过比较大小关系来找到适合插入的叶子节点
(4)一定要有root->left (root->right)去接住返回的新节点,不然就连接不成功了。
(5)返回root
2、代码如下:
class Solution {
public:
TreeNode* insertIntoBST(TreeNode* root, int val) {
if (root == NULL) {
return new TreeNode(val);
}
if (val < root->val) {
if (root->left == NULL) {
root->left = new TreeNode(val);
} else {
root->left = insertIntoBST(root->left, val);
}
} else {
if (root->right == NULL) {
root->right = new TreeNode(val);
} else {
root->right = insertIntoBST(root->right, val);
}
}
return root;
}
};
这里其中的很多细节可以省略,简洁版如下:
class Solution {
public:
TreeNode* insertIntoBST(TreeNode* root, int val) {
if (root == NULL) {
// 添加一个节点,返回上一层,就完成了父子结点的操作了
TreeNode* kk = new TreeNode(val);
return kk;
}
// 找到适合元素插入的位置
if (root->val >val) {
// 接住节点
root->left = insertIntoBST(root->left, val);
} else {
// 接住节点
root->right = insertIntoBST(root->right, val);
}
return root;
}
};
四、删除二叉搜索树中的节点
1、思路:
这道题分为五种情况,
(1)删除叶子节点,左右都为空;
(2)左不为空,右为空;
(3)右为空,左不为空;
(4)左不为空,右不为空(需要大幅度调整树的结构,如果以删除节点的右子树来顶替的话,就需要找右边最小的节点来接被删除结点的左子树;
(5)没有找到;
采用的是递归,采用前序遍历
2、代码如下:
class Solution {
public:
TreeNode* deleteNode(TreeNode* root, int key) {
// 情况一:没找到节点
if (root == NULL) return root;
if (root->val == key) {
// 情况二:找到的是叶子节点
if (root->left == nullptr && root->right == nullptr) {
delete root;
return nullptr;
}
// 情况三:左不为空,右为空
else if (root->left == nullptr) {
auto temp = root->right;
delete root;
return temp;
}
// 情况四:右为空,左不为空
else if (root->right == nullptr) {
auto temp = root->left;
delete root;
return temp;
}
// 情况四:左右都不为空
else{
// 从右子树找
auto cur = root->right;
// 找右子树最左边的值,这个值最小,接住root左子树
while (cur->left != nullptr) {
cur = cur->left;
}
// 把左子树接到右子树的最左边
cur->left = root->left;
// 方便删除
TreeNode* temp = root;
// 连接右子树
root = root->right;
delete temp;
return root;
}
}
// 左遍历,用root接住
if (key < root->val) root->left = deleteNode(root->left, key);
if (key > root->val) root->right = deleteNode(root->right, key);
return root;
}
};
学习时间:2小时
leave message:
Courage is like love;it must have hope for nourishment.
勇气就像爱,需要希望来滋养。