【刷爆LeetCode】五月算法集训（20）二叉搜索树

题目来源于知识星球—英雄算法联盟，五月算法集训专题

前言

跟随英雄算法联盟博主—英雄哪里出来，每天完成相应的算法练习，一个月后，必定会有所成长！

---

一、700.二叉搜索树中的搜索（简单）

1.题目描述

2.解题思路

在有序的二叉树中，如果根节点为空，则返回空；如果值小于当前根节点的值，则去左子树查找；
如果值大于当前根节点的值，则去右子树查找；如果等于，直接返回。

3.代码演示（C++）

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

    }
};

4.题目链接

题目传送门

---

二、230.二叉搜索树中第K小的元素（中等）

1.题目描述

2.解题思路

对二叉树进行中序遍历后得到有序数组，访问节点，k递减，当k等于0时，对应的节点为第k小的节点。

3.代码演示（C++）

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

4.题目链接

题目传送门

---

三、108.将有序数组转换为二叉搜索树（简单）

1.题目描述

2.解题思路

对于给定的有序数组，选择中点值作为根节点，左边的数都位于根节点的左侧，并且它的根节点是当前
根节点的左儿子，同理，右边的数都位于根节点的右侧，并且它的根节点是当前根节点的右儿子。

3.代码演示（C++）

class Solution 
{
    TreeNode *dfs(vector<int>& nums,int l,int r)
    {
        if(l>r)
        {
            return nullptr;
        }
        int mid=(l+r)>>1;
        TreeNode *n=new TreeNode(nums[mid]);
        n->left=dfs(nums,l,mid-1);
        n->right=dfs(nums,mid+1,r);
        return n;
    }
public:
    TreeNode* sortedArrayToBST(vector<int>& nums) 
    {
        int l=0;
        int r=nums.size()-1;
        return dfs(nums,l,r);
    }
};

4.题目链接

题目传送门

---

四、1382.将二叉搜索树变平衡（中等）

1.题目描述

2.解题思路

将二叉搜索树中序遍历得到有序数组，然后再调用第三题的接口即可。

3.代码演示（C++）

class Solution 
{
    vector<TreeNode*>ret;
    TreeNode* dfs(vector<TreeNode*>& nodes,int l,int r)
    {
        if(l>r)
        {
            return nullptr;
        }
        int mid=(l+r)>>1;
        TreeNode *n=nodes[mid];
        n->left=dfs(nodes,l,mid-1);
        n->right=dfs(nodes,mid+1,r);
        return n;
    }
    TreeNode* sortedArrayToBST(vector<TreeNode*>& nodes)
    {
        int l=0;
        int r=nodes.size()-1;
        return dfs(nodes,l,r);
    }
    void inorder(TreeNode* root)
    {
        if(root)
        {
            inorder(root->left);
            ret.push_back(root);
            inorder(root->right);
        }
    }
public:
    TreeNode* balanceBST(TreeNode* root) 
    {
        ret.clear();
        inorder(root);
        return sortedArrayToBST(ret);
    }
};

4.题目链接

题目传送门

---

总结

每天跟随英雄哥学习相关的算法，一个月会收获很多，如果你想了解更多关于知识星球的内容，欢迎联系我！