235. 二叉搜索树的最近公共祖先
题目:
给定一个二叉搜索树, 找到该树中两个指定节点的最近公共祖先。
百度百科中最近公共祖先的定义为:“对于有根树 T 的两个结点 p、q,最近公共祖先表示为一个结点 x,满足 x 是 p、q 的祖先且 x 的深度尽可能大(一个节点也可以是它自己的祖先)。”
例如,给定如下二叉搜索树: root = [6,2,8,0,4,7,9,null,null,3,5]
示例 1:
输入: root = [6,2,8,0,4,7,9,null,null,3,5], p = 2, q = 8 输出: 6 解释: 节点2
和节点8
的最近公共祖先是6。
示例 2:
输入: root = [6,2,8,0,4,7,9,null,null,3,5], p = 2, q = 4 输出: 2 解释: 节点2
和节点4
的最近公共祖先是2
, 因为根据定义最近公共祖先节点可以为节点本身。
说明:
- 所有节点的值都是唯一的。
- p、q 为不同节点且均存在于给定的二叉搜索树中。
思路:
要在二叉搜索树中找到两个节点的最近公共祖先,可以利用二叉搜索树的性质:对于任何节点 n
,其左子树上的所有节点的值都小于 n
,右子树上的所有节点的值都大于 n
。
因此,通过比较节点 p
和 q
的值与当前节点 root
的值,可以确定最近公共祖先的位置。
- 如果
p
和q
的值都小于当前节点root
的值,那么最近公共祖先一定在当前节点的左子树中,因此可以递归地在左子树中查找。 - 如果
p
和q
的值都大于当前节点root
的值,那么最近公共祖先一定在当前节点的右子树中,因此可以递归地在右子树中查找。 - 如果
p
和q
分别位于当前节点的两侧(一个在左,一个在右),或者其中一个节点就是当前节点本身,那么当前节点就是它们的最近公共祖先。
上代码:
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
// 如果p和q都小于当前节点,说明最近公共祖先在左子树
if (p->val < root->val && q->val < root->val) {
return lowestCommonAncestor(root->left, p, q);
}
// 如果p和q都大于当前节点,说明最近公共祖先在右子树
if (p->val > root->val && q->val > root->val) {
return lowestCommonAncestor(root->right, p, q);
}
// 如果p和q分居当前节点的两侧,或者其中一个就是当前节点
return root;
}
};
701.二叉搜索树中的插入操作
题目:
给定二叉搜索树(BST)的根节点 root
和要插入树中的值 value
,将值插入二叉搜索树。 返回插入后二叉搜索树的根节点。 输入数据 保证 ,新值和原始二叉搜索树中的任意节点值都不同。
注意,可能存在多种有效的插入方式,只要树在插入后仍保持为二叉搜索树即可。 你可以返回 任意有效的结果 。
示例 1:
输入:root = [4,2,7,1,3], val = 5 输出:[4,2,7,1,3,5] 解释:另一个满足题目要求可以通过的树是:
示例 2:
输入:root = [40,20,60,10,30,50,70], val = 25 输出:[40,20,60,10,30,50,70,null,null,25]
示例 3:
输入:root = [4,2,7,1,3,null,null,null,null,null,null], val = 5 输出:[4,2,7,1,3,5]
提示:
- 树中的节点数将在
[0, 104]
的范围内。 -108 <= Node.val <= 108
- 所有值
Node.val
是 独一无二 的。 -108 <= val <= 108
- 保证
val
在原始BST中不存在。
思路:
要在二叉搜索树(BST)中插入一个新的值,我们可以利用二叉搜索树的性质:左子树中的所有节点值都小于根节点,右子树中的所有节点值都大于根节点。
因此,插入一个值的过程就是不断比较新值与当前节点的值,并决定是进入左子树还是右子树,直到找到合适的插入位置。
- 从根节点开始,比较新值
val
与当前节点的值:- 如果
val
小于当前节点的值,则进入左子树继续比较; - 如果
val
大于当前节点的值,则进入右子树继续比较。
- 如果
- 当找到一个空的子节点位置时,将新节点插入到这个位置。
- 如果树为空,直接返回一个包含新值的节点作为根节点。
上代码:
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode() : val(0), left(nullptr), right(nullptr) {}
* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
* TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
* };
*/
class Solution {
public:
TreeNode* insertIntoBST(TreeNode* root, int val) {
// 如果当前节点为空,直接创建新节点
if (root == nullptr) {
return new TreeNode(val);
}
// 比较新值与当前节点的值,决定向左或向右插入
if (val < root->val) {
// 插入左子树
root->left = insertIntoBST(root->left, val);
} else {
// 插入右子树
root->right = insertIntoBST(root->right, val);
}
// 返回更新后的根节点
return root;
}
};
450.删除二叉搜索树中的节点
题目:
给定一个二叉搜索树的根节点 root 和一个值 key,删除二叉搜索树中的 key 对应的节点,并保证二叉搜索树的性质不变。返回二叉搜索树(有可能被更新)的根节点的引用。
一般来说,删除节点可分为两个步骤:
- 首先找到需要删除的节点;
- 如果找到了,删除它。
示例 1:
输入:root = [5,3,6,2,4,null,7], key = 3 输出:[5,4,6,2,null,null,7] 解释:给定需要删除的节点值是 3,所以我们首先找到 3 这个节点,然后删除它。 一个正确的答案是 [5,4,6,2,null,null,7], 如下图所示。 另一个正确答案是 [5,2,6,null,4,null,7]。
示例 2:
输入: root = [5,3,6,2,4,null,7], key = 0 输出: [5,3,6,2,4,null,7] 解释: 二叉树不包含值为 0 的节点
示例 3:
输入: root = [], key = 0 输出: []
提示:
- 节点数的范围
[0, 104]
. -105 <= Node.val <= 105
- 节点值唯一
root
是合法的二叉搜索树-105 <= key <= 105
思路:
要在二叉搜索树中删除一个节点,并保持树的二叉搜索树性质不变,可以分为以下几种情况来处理:
- 目标节点是叶子节点:直接删除这个节点即可。
- 目标节点只有一个子节点:删除该节点,并让其子节点替代它的位置。
- 目标节点有两个子节点:找到该节点的中序后继(即右子树中的最小节点),用这个中序后继替代该节点的值,然后删除中序后继节点。
上代码:
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode() : val(0), left(nullptr), right(nullptr) {}
* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
* TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
* };
*/
class Solution {
public:
TreeNode* deleteNode(TreeNode* root, int key) {
if (root == nullptr) return nullptr;
// 寻找要删除的节点
if (key < root->val) {
// 去左子树中寻找
root->left = deleteNode(root->left, key);
} else if (key > root->val) {
// 去右子树中寻找
root->right = deleteNode(root->right, key);
} else {
// 找到目标节点
if (root->left == nullptr) {
// 节点只有右子树或无子树
TreeNode* temp = root->right;
delete root; // 释放内存
return temp;
} else if (root->right == nullptr) {
// 节点只有左子树
TreeNode* temp = root->left;
delete root; // 释放内存
return temp;
} else {
// 节点有两个子树,找到右子树中的最小节点(中序后继)
TreeNode* minNode = findMin(root->right);
// 用中序后继的值替代当前节点
root->val = minNode->val;
// 删除中序后继节点
root->right = deleteNode(root->right, minNode->val);
}
}
return root;
}
private:
// 查找二叉树的最小节点
TreeNode* findMin(TreeNode* node) {
while (node->left != nullptr) {
node = node->left;
}
return node;
}
};