【五月集训5.20】———二叉搜索树

☘前言☘

开更五月集训专题，由浅入深，深入浅出，飞向大厂！

🧑🏻作者简介：一个从工业设计改行学嵌入式的年轻人
✨联系方式：2201891280(QQ)
⏳全文大约阅读时间： 20min

---

---

700. 二叉搜索树中的搜索

700. 二叉搜索树中的搜索

解题思路

利用二叉树的特性，大于就搜索右子树，否则搜索左子树。如果是空树直接返回。

代码

class Solution {
public:
    TreeNode* searchBST(TreeNode* root, int val) {
        if(root && root -> val == val)  return root;
        else if(root && root->val > val)    return searchBST(root->left, val);
        else if(root && root->val < val)    return searchBST(root->right, val);
        else return root;
    }
};

注意的点

1. 直接就是前序遍历 没啥意思

---

230. 二叉搜索树中第K小的元素

230. 二叉搜索树中第K小的元素

解题思路

直接就是中序遍历，然后到对应的元素的时候记录一下就好了。

代码

class Solution {
    void dfs(TreeNode* root, int k,int &i,int & ret){
        if(root->left)  dfs(root->left,k,i,ret);
        ++i;
        if(i == k)  ret = root->val;
        else if(root->right) dfs(root->right,k,i,ret);
    }
public:
    int kthSmallest(TreeNode* root, int k) {
        int i = 0,ret = 0;
        dfs(root,k,i,ret);
        return ret;
    }
};

注意的点

1. 没啥，这个不难。

---

108. 将有序数组转换为二叉搜索树

108. 将有序数组转换为二叉搜索树

解题思路

利用性质，每次选取中间元素建树就好了。

代码

class Solution {
    TreeNode *jianli(vector<int> &nums,int left,int right){
        if(left > right)    return NULL;
        int mid = (right + left) / 2;
        TreeNode* ret = new TreeNode(nums[mid],jianli(nums,left,mid - 1),jianli(nums,mid+1,right));
        return ret;
    }
public:
    TreeNode* sortedArrayToBST(vector<int>& nums) {
        int n = nums.size();
        return jianli(nums,0,n - 1);
    }
};

注意的点

因为向下取整，如果中间的是两个元素，选择的会是小的，当然也无所谓，虽然和答案不一样但是可以ac。如果想和标准答案一样那就需要(left + right + 1)/2

---

1382. 将二叉搜索树变平衡

1382. 将二叉搜索树变平衡

解题思路

本来是需要调整的，什么左旋右旋乱七八糟的，但是呢，我们偷懒，利用第二题和第三题的函数接口，直接保存结果到数组，然后重新建一棵树就完事了。

代码

class Solution {
   void dfs(TreeNode* root, vector<int> &ret){
        if(root->left)  dfs(root->left,ret);
        ret.push_back(root->val);
        if(root->right) dfs(root->right,ret);
    }
    TreeNode *jianli(vector<int> &nums,int left,int right){
        if(left > right)    return NULL;
        int mid = (right + left) / 2;
        TreeNode* ret = new TreeNode(nums[mid],jianli(nums,left,mid - 1),jianli(nums,mid+1,right));
        return ret;
    }
public:
    TreeNode* balanceBST(TreeNode* root) {
        vector<int> tmp;
        dfs(root,tmp);
        return jianli(tmp,0,tmp.size() - 1);
    }
};

注意的点

没啥

---

写在最后

大家加油！