二叉树
235. 二叉搜索树的最近公共祖先
题目描述
给定一个二叉搜索树, 找到该树中两个指定节点的最近公共祖先。
百度百科中最近公共祖先的定义为:“对于有根树 T 的两个结点 p、q,最近公共祖先表示为一个结点 x,满足 x 是 p、q 的祖先且 x 的深度尽可能大(一个节点也可以是它自己的祖先)。”
例如,给定如下二叉搜索树: root = [6,2,8,0,4,7,9,null,null,3,5]
思路
二叉搜索树,可以通过比较当前节点值和目标值的大小,将搜索范围缩小到只搜索左或右子树;
对于下层返回的节点,需要接住,根据处理结果判断是否继续向上传。
代码
/**
* 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 == NULL) return NULL;
if (root->val > p->val && root->val > q->val)
{
TreeNode* left = lowestCommonAncestor(root->left, p, q);
if (left != NULL) return left;
}
if (root->val < p->val && root->val < q->val)
{
TreeNode* right = lowestCommonAncestor(root->right, p ,q);
if (right != NULL) return right;
}
return root;
}
};
701. 二叉搜索树中的插入操作
题目描述
给定二叉搜索树(BST)的根节点 root 和要插入树中的值 value ,将值插入二叉搜索树。 返回插入后二叉搜索树的根节点。 输入数据 保证 ,新值和原始二叉搜索树中的任意节点值都不同。
注意,可能存在多种有效的插入方式,只要树在插入后仍保持为二叉搜索树即可。 你可以返回 任意有效的结果 。
思路
比较根节点和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 == NULL)
{
TreeNode* node = new TreeNode(val);
return node;
}
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 对应的节点,并保证二叉搜索树的性质不变。返回二叉搜索树(有可能被更新)的根节点的引用。
一般来说,删除节点可分为两个步骤:
首先找到需要删除的节点;
如果找到了,删除它。
思路
找到要删除的节点,看在以该节点为根的子树中,根据左子树、右子树是否为空,分为四种情况:
(1)左子树空,右子树不空,修剪后的子树即为原本子树的右子树;
(2)左子树不空,右子树空,修剪后的子树即为原本子树的左子树;
(3)左右子树均不空,把左子树嫁接到右子树最左节点的左侧,即右子树为需要返回的修剪后的子树。
代码
/**
* 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* traverse(TreeNode* node, int key)
{
if (node == NULL) return node;
if (node->val == key)
{
if (node->left == NULL) return node->right;
else if (node->right == NULL) return node->left;
else{
TreeNode* cur = node->right;
while(cur->left != NULL)
{
cur= cur->left;
}
TreeNode* tmp = node;
cur->left = node->left;
node = node->right;
delete tmp;
return node;
}
}
if (node->val > key) node->left = traverse(node->left, key);
if (node->val < key) node->right = traverse(node->right, key);
return node;
}
TreeNode* deleteNode(TreeNode* root, int key) {
root = traverse(root, key);
return root;
}
};