力扣数据结构14天学习计划day13

力扣700

力扣700—二叉搜索树中的搜索

题意

给定二叉搜索树（BST）的根节点 root 和一个整数值 val。

你需要在 BST 中找到节点值等于 val 的节点。 返回以该节点为根的子树。 如果节点不存在，则返回 null 。

法1—递归

二叉搜索树满足如下性质：

• 左子树所有节点的元素值均小于根的元素值；
• 右子树所有节点的元素值均大于根的元素值。

据此可以得到如下算法：

• 若 root 为空则返回空节点；
• 若val=root.val，则返回root；
• 若 val<root.val，递归左子树；
• 若 val>root.val，递归右子树。

class Solution 
{
public:
    
    TreeNode* searchBST(TreeNode* root, int val) 
    {
        if(root==nullptr)
            return nullptr;
        if(root->val==val)
            return root;
        //开始根据val的值，进行递归
        return searchBST(val>root->val?root->right:root->left,val);
    }
};

法2—迭代

我们将法一的递归改成迭代写法： 

• 若 root 为空则跳出循环，并返回空节点；
• 若 val=root.val，则返回 root；
• 若 val<root.val，将 root 置为root->left；
• 若 val>root.val，将 root 置为 root->right。

 

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

        //开始根据val的值，进行递归
        //return searchBST(val>root->val?root->right:root->left,val);

        while(root)
        {
            if(root->val==val)
                return root;
            else if(val>root->val)
                root=root->right;
            else
                root=root->left;
        }
        return nullptr;
    }
};

力扣701—二叉搜索树中的插入操作

题意

给定二叉搜索树（BST）的根节点 root 和要插入树中的值 value ，将值插入二叉搜索树。 返回插入后二叉搜索树的根节点。 输入数据 保证 ，新值和原始二叉搜索树中的任意节点值都不同。

注意，可能存在多种有效的插入方式，只要树在插入后仍保持为二叉搜索树即可。 你可以返回 任意有效的结果 。

 

 解题思路—模拟

当将 val 插入到以root 为根的子树上时，根据 val 与root->val 的大小关系，就可以确定要将 val 插入到哪个子树中。

• 如果该子树不为空，则问题转化成了将 \textit{val}val 插入到对应子树上。
• 否则，该子树为空，在此处新建一个以val 为值的节点，并链接到其父节点root 上。

class Solution
{
public:
    TreeNode* insertIntoBST(TreeNode* root, int val) 
    {
        if(nullptr==root)
            return new TreeNode(val);
        TreeNode* node=root;
        while(node)
        {
            // 插入的值比当前节点大，则应该往当前节点的右孩子走
            if(val>node->val)
            {
                //如果当前节点的右孩子为空了，就将该节点插入在该节点的右孩子处
                if(node->right==nullptr)
                {
                    node->right=new TreeNode(val);
                    break;
                }
                //当前节点的右孩子不空，则将node更新为它的右孩子，让插入值和node的右孩子相比
                // 看是插在 node右孩子的左孩子处呢，还是插在node右孩子的右孩子处
                else
                {
                    node=node->right;
                }
            }
            // 插入的值比当前节点小，则应该往当前节点的左孩子走
            else
            {
                if(node->left==nullptr)
                {
                    node->left=new TreeNode(val);
                    break;
                }
                else
                {
                    node=node->left;
                }
            }
        }
        return root;
    }
};

解法2—递归

class Solution
{
public:
    TreeNode* insertIntoBST(TreeNode* root, int val) 
    {
        if(nullptr==root)
            return new TreeNode(val);

        /*
            //迭代
        TreeNode* node=root;
        while(node)
        {
            // 插入的值比当前节点大，则应该往当前节点的右孩子走
            if(val>node->val)
            {
                //如果当前节点的右孩子为空了，就将该节点插入在该节点的右孩子处
                if(node->right==nullptr)
                {
                    node->right=new TreeNode(val);
                    break;
                }
                //当前节点的右孩子不空，则将node更新为它的右孩子，让插入值和node的右孩子相比
                // 看是插在 node右孩子的左孩子处呢，还是插在node右孩子的右孩子处
                else
                {
                    node=node->right;
                }
            }
            // 插入的值比当前节点小，则应该往当前节点的左孩子走
            else
            {
                if(node->left==nullptr)
                {
                    node->left=new TreeNode(val);
                    break;
                }
                else
                {
                    node=node->left;
                }
            }
        }
        return root;
        */

        //递归
        if(val>root->val)
        {
            root->right=insertIntoBST(root->right,val);
        }
        else
        {
            root->left=insertIntoBST(root->left,val);
        }
        return root;
    }
};