【LeetCode】235. 二叉搜索树的最近公共祖先
题意:给定一个二叉搜索树, 找到该树中两个指定节点的最近公共祖先。
百度百科中最近公共祖先的定义为:“对于有根树 T 的两个结点 p、q,最近公共祖先表示为一个结点 x,满足 x 是 p、q 的祖先且 x 的深度尽可能大(一个节点也可以是它自己的祖先)。”
思路:相比于普通二叉树,二叉搜索树要思考利用有序性。判断二叉搜索树上两个结点的最近公共祖先,实际上就是自上而下遍历时满足结点值在这两个结点区间之内的第一个点。
代码:
/**
* 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) {
if (!root || root == p || root == q)
{
return root;
}
if ((root->val < p->val && root->val > q->val)
|| (root->val > p->val && root->val < q->val))
{
return root;
}
else if (root->val < p->val && root->val < q->val)
{
return lowestCommonAncestor(root->right, p, q);
}
else
{
return lowestCommonAncestor(root->left, p, q);
}
}
};
【LeetCode】701.二叉搜索树中的插入操作
题意:给定二叉搜索树(BST)的根节点 root 和要插入树中的值 value ,将值插入二叉搜索树。 返回插入后二叉搜索树的根节点。 输入数据 保证 ,新值和原始二叉搜索树中的任意节点值都不同。
注意,可能存在多种有效的插入方式,只要树在插入后仍保持为二叉搜索树即可。 你可以返回 任意有效的结果 。
思路:递归判断需要插入的值,小就往左子树,大就往右子树,遇到叶子节点时赋值。
代码:
/**
* 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)
{
return new TreeNode(val);
}
if (val > root->val)
{
root->right = insertIntoBST(root->right, val);
}
else
{
root->left = insertIntoBST(root->left, val);
}
return root;
}
};
【LeetCode】450.删除二叉搜索树中的节点
题意:给定一个二叉搜索树的根节点 root 和一个值 key,删除二叉搜索树中的 key 对应的节点,并保证二叉搜索树的性质不变。返回二叉搜索树(有可能被更新)的根节点的引用。
一般来说,删除节点可分为两个步骤:
首先找到需要删除的节点;
如果找到了,删除它。
思路:递归二叉搜索树,分情况讨论,如果当前结点值比key小则往右子树找,比key大则往左子树找。找到key相同时,判断如果是叶子结点就直接删除返回,如果左子树为空则返回右子树,如果右子树为空则返回左子树,如果左右子树都不为空则将左子树作为右子树最左边结点的左子树(比它大的最小值位置)。
代码:
/**
* 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)
{
return NULL;
}
if (key == root->val)
{
// 左右子树都为空
if (!root->left && !root->right)
{
delete root;
return NULL;
}
// 左子树为空
else if (!root->left)
{
return root->right;
}
// 右子树为空
else if (!root->right)
{
return root->left;
}
// 左右子树都不为空
else
{
TreeNode* right = root->right;
while (right->left)
{
right = right->left;
}
right->left = root->left;
TreeNode* node = root;
root = root->right;
delete node;
node = NULL;
return root;
}
}
else if (key > root->val)
{
root->right = deleteNode(root->right, key);
}
else
{
root->left = deleteNode(root->left, key);
}
return root;
}
};
心态:“第六章 二叉树 part08” 拿下!
参考资料