二叉搜索树中的最近公共祖先
二叉树的最近公共祖先,其实也可以用寻找二叉树的最近公共祖先的方法来找,但是要善于利用二叉搜索树的特性,我们想要的是p、q的祖先,那么这个祖先一定在[p,q]或者[q,p]之间,而二叉搜索树具有一个节点一定大于它左子树上节点小于它右子树节点的特性,因此我们第一次找到节点满足这个区间的那个节点就是公共祖先。
class Solution {
public:
TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
if(root->val > q->val && root->val > p->val){
return lowestCommonAncestor(root->left,p,q);
}else if(root->val < q->val && root->val < p->val){
return lowestCommonAncestor(root->right,p,q);
}else{
return root;
}
}
};
二叉搜索树中的插入操作
这道题需要我们进行插入操作,原本树是一个二叉树,插入之后也是一个二叉树,我们只需要在原有的二叉树的空节点插入正确的数值,就可以使得原来的二叉搜索树变成一棵新的二叉搜索树。
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;
}
};
删除二叉搜索树中的节点
删除二叉搜索树的节点首先需要找到那个节点,然后分情况删除节点,其中最复杂的情况是要删除的节点既有左孩子又有右孩子,我们首先需要将要删除的节点的左子树接到其右节点的最左边的那个节点上,然后将这个右节点补上删除节点的位置。
class Solution {
public:
TreeNode* deleteNode(TreeNode* root, int key) {
if(root == NULL) return NULL;
if(root->val == key)
{
if(!root->left && !root->right)
{
delete root;
return NULL;
}else if(root->left && !root->right)
{
auto retNode = root->left;
delete root;
return retNode;
}else if(!root->left && root->right)
{
auto retNode = root->right;
delete root;
return retNode;
}else{
TreeNode* node = root->right;
while(node->left)
{
node = node->left;
}
node->left = root->left;
auto retNode = root->right;
delete root;
return retNode;
}
}
if(root->val > key) root->left = deleteNode(root->left,key);
if(root->val < key) root->right = deleteNode(root->right,key);
return root;
}
};