代码随想录算法训练营第32天|LeetCode701.二叉搜索树中的插入操作

代码随想录算法训练营第32天|LeetCode701.二叉搜索树中的插入操作

1、LeetCode701.二叉搜索树中的插入操作

https://leetcode.cn/problems/insert-into-a-binary-search-tree/

请添加图片描述

第一想法:有点复杂,感觉会涉及到旋转之类的操作。一直遍历找到能插入的位置就插入。

递归法

可以不考虑题目中提示所说的改变树的结构的插入方式。
只要遍历二叉搜索树,找到空节点 插入元素就可以了,那么这道题其实就简单了。

  1. 参数和返回值
    TreeNode* insertIntoBST(TreeNode* root, int val)
    或无返回值——更复杂,需要额外定义一个parent变量。
  2. 终止条件
    找到节点为空,插入
  3. 单层逻辑
    不需要遍历整棵树

代码(有返回值)

犯错:插入时没有赋值给root->left

class Solution {
public:
    TreeNode* insertIntoBST(TreeNode* root, int val) {
        // 递归1 有返回值
        if(root == NULL){
            TreeNode* node = new TreeNode(val);
            return node;
        }
        if(root->val > val){
            // 往左边插入
            root->left = insertIntoBST(root->left, val); //注意这里要赋值给root->left
        }
        if(root->val < val){
            // 往右边插入
            root->right = insertIntoBST(root->right, val);
        }
        // 插入后返回树
        return root;
    }
};

代码(没有返回值)
需要额外记录parent变量,并及时更新。

class Solution {
public:
    // 递归2 没有返回值
    TreeNode* parent;//记录父节点
    void traversal(TreeNode* root, int val){
        if(root == NULL){
            TreeNode* node = new TreeNode(val);
            if(parent->val < val)
                parent->right = node;
            else
                parent->left = node;
            return ; 
        }
        parent = root;//及时更新
        if(root->val < val)
            traversal(root->right, val);
        else
            traversal(root->left, val);
        return ;
    }
    TreeNode* insertIntoBST(TreeNode* root, int val) {
        // 递归2 没有返回值
        if(root == NULL){
            TreeNode* node = new TreeNode(val);
            return node;
        }
        traversal(root, val);
        return root;

    }
};

迭代法

需要使用pre和cur。这一题中parent即cur

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    TreeNode* insertIntoBST(TreeNode* root, int val) {
        // 法3 迭代法
        if(root == NULL){
            TreeNode* node = new TreeNode(val);
            return node;
        }
        TreeNode* parent = NULL;
        TreeNode* cur = root;
        while(cur){
            parent = cur;//注意这里及时更新parent
            if(cur->val < val)
                cur = cur->right;
            else   
                cur = cur->left;
        }
        TreeNode* node = new TreeNode(val);
        if(parent->val < val)
            parent->right = node;
        else
            parent->left = node;
        return root;
    }
};
  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
第二十二算法训练营主要涵盖了Leetcode题目的三道题目,分别是Leetcode 28 "Find the Index of the First Occurrence in a String",Leetcode 977 "有序数组的平方",和Leetcode 209 "长度最小的子数组"。 首先是Leetcode 28题,题目要求在给定的字符串找到第一个出现的字符的索引。思路是使用双指针来遍历字符串,一个指向字符串的开头,另一个指向字符串的结尾。通过比较两个指针所指向的字符是否相等来判断是否找到了第一个出现的字符。具体实现的代码如下: ```python def findIndex(self, s: str) -> int: left = 0 right = len(s) - 1 while left <= right: if s[left == s[right]: return left left += 1 right -= 1 return -1 ``` 接下来是Leetcode 977题,题目要求对给定的有序数组的元素进行平方,并按照非递减的顺序返回结果。这里由于数组已经是有序的,所以可以使用双指针的方法来解决问题。一个指针指向数组的开头,另一个指针指向数组的末尾。通过比较两个指针所指向的元素的绝对值的大小来确定哪个元素的平方应该放在结果数组的末尾。具体实现的代码如下: ```python def sortedSquares(self, nums: List[int]) -> List[int]: left = 0 right = len(nums) - 1 ans = [] while left <= right: if abs(nums[left]) >= abs(nums[right]): ans.append(nums[left ** 2) left += 1 else: ans.append(nums[right ** 2) right -= 1 return ans[::-1] ``` 最后是Leetcode 209题,题目要求在给定的数组找到长度最小的子数组,使得子数组的和大于等于给定的目标值。这里可以使用滑动窗口的方法来解决问题。使用两个指针来表示滑动窗口的左边界和右边界,通过移动指针来调整滑动窗口的大小,使得滑动窗口的元素的和满足题目要求。具体实现的代码如下: ```python def minSubArrayLen(self, target: int, nums: List[int]) -> int: left = 0 right = 0 ans = float('inf') total = 0 while right < len(nums): total += nums[right] while total >= target: ans = min(ans, right - left + 1) total -= nums[left] left += 1 right += 1 return ans if ans != float('inf') else 0 ``` 以上就是第二十二算法训练营的内容。通过这些题目的练习,可以提升对双指针和滑动窗口等算法的理解和应用能力。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值