235. 二叉搜索树的最近公共祖先
/*
如果小于,答案一定在右子树,直接返回右子树的答案;反之返回左子树的答案
*/
TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
if(root==nullptr) return nullptr;
//小于
if(root->val<p->val&&root->val<q->val){
return lowestCommonAncestor(root->right,p,q);
}
//大于
if(root->val>p->val&&root->val>q->val){
return lowestCommonAncestor(root->left,p,q);
}
return root;
}
701. 二叉搜索树中的插入操作
/*
整体思路:遍历找到空位置就插。
返回值:保证最后的返回值是正确的就可以了。
*/
TreeNode* insertIntoBST(TreeNode* root, int val) {
if(root==nullptr) return new TreeNode(val);
if(val < root->val){
if(root->left==nullptr) root->left = new TreeNode(val);
else insertIntoBST(root->left,val);
}else if(val > root->val){
if(root->right==nullptr) root->right = new TreeNode(val);
else insertIntoBST(root->right,val);
}
return root;
}
450. 删除二叉搜索树中的节点
这里就把二叉搜索树中删除节点遇到的情况都搞清楚。
有以下五种情况:
第一种情况:没找到删除的节点,遍历到空节点直接返回了
找到删除的节点
第二种情况:左右孩子都为空(叶子节点),直接删除节点, 返回NULL为根节点
第三种情况:删除节点的左孩子为空,右孩子不为空,删除节点,右孩子补位,返回右孩子为根节点
第四种情况:删除节点的右孩子为空,左孩子不为空,删除节点,左孩子补位,返回左孩子为根节点
第五种情况:左右孩子节点都不为空,则将删除节点的左子树头结点(左孩子)放到删除节点的右子树的最左面节点的左孩子上,返回删除节点右孩子为新的根节点。
TreeNode* deleteNode(TreeNode* root, int key) {
if(root == nullptr) return root;
if(key == root->val){
//左子树右子树均为空。
if(root->left == nullptr&& root->right == nullptr){
delete root;
return nullptr;
}
//左子树为空,右子树不为空。
else if(root->left == nullptr&& root->right != nullptr){
auto ret = root->right;
delete root;
return ret;
}
//右子树为空,左子树不为空。
else if(root->left != nullptr&& root->right == nullptr){
auto ret = root->left;
delete root;
return ret;
}
//左右子树均不为空。
else if(root->left != nullptr&& root->right != nullptr){
//先遍历找到右子树的最左节点
auto cur = root->right;
while(cur->left!=nullptr){
cur = cur->left;
}
cur->left = root->left;
auto ret = root->right;
delete root;
return ret;
}
}
else if(key < root->val) root->left = deleteNode(root->left,key);
else if(key > root->val) root->right = deleteNode(root->right,key);
return root;
}
669. 修剪二叉搜索树
TreeNode* trimBST(TreeNode* root, int low, int high) {
if (root == nullptr) return nullptr;
if (root->val < low) return trimBST(root->right, low, high);
if (root->val > high) return trimBST(root->left, low, high);
root->left = trimBST(root->left, low, high);
root->right = trimBST(root->right, low, high);
return root;
}
108. 将有序数组转换为二叉搜索树
private:
TreeNode* travelsal(vector<int>& nums, int left, int right){
if(left > right) return nullptr;
int mid = left + (right - left)/2;
auto root = new TreeNode(nums[mid]);
root->left = travelsal(nums,left,mid-1);
root->right = travelsal(nums,mid+1,right);
return root;
}
public:
TreeNode* sortedArrayToBST(vector<int>& nums) {
return travelsal(nums,0,nums.size()-1);
}