day1|704. 二分查找、27. 移除元素

76 阅读1分钟

704. 二分查找

题目:给定一个 n 个元素有序的(升序)整型数组 nums 和一个目标值 target ,写一个函数搜索 nums 中的 target,如果目标值存在返回下标,否则返回 -1。

示例 1:

输入: nums = [-1,0,3,5,9,12], target = 9
输出: 4
解释: 9 出现在 nums 中并且下标为 4

重点:边界的处理规则,是根据区间的定义来写的

class Solution:
    def search(self, nums: List[int], target: int) -> int:
    # 索引定义为左闭右闭区间
        left = 0
        right = len(nums)-1
        
        while left <= right:  # 闭区间,所以left和right可以相等
            mid = (left+right)//2
            if nums[mid] < target:
                left = mid + 1 # 因为target已经和mid比过了,因此更新后的区间应不包含mid
            elif nums[mid] > target:
                right = mid -1  # 同上
            else:
                return mid
        
        return -1

27. 移除元素

给你一个数组 nums 和一个值 val,你需要 原地 移除所有数值等于 val 的元素,并返回移除后数组的新长度。

不要使用额外的数组空间,你必须仅使用 O(1) 额外空间并原地修改输入数组。

元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。

示例 1: 给定 nums = [3,2,2,3], val = 3,函数应该返回新的长度 2, 
并且 nums 中的前两个元素均为 2。 
你不需要考虑数组中超出新长度后面的元素。

示例 2: 给定 nums = [0,1,2,2,3,0,4,2],val = 2, 函数应该返回新的长度 5,
并且 nums 中的前五个元素为 0, 1, 3, 0, 4。

思路: 使用慢指针记录有效数组的长度,快指针便利数组,寻找有效数组的元素。

class Solution:
    def removeElement(self, nums: List[int], val: int) -> int:
        slow = 0 
        
        for fast in range(len(nums)):
            if nums[fast] != val:
                nums[fast],nums[slow] = nums[slow],nums[fast]
                slow += 1
        
        return slow