【lc刷题】35/34 搜索插入位置/在排序数组中查找元素的第一个和最后一个位置(143-144/300)

143-144/300

35. 搜索插入位置
     
    给定一个排序数组和一个目标值，在数组中找到目标值，并返回其索引。如果目标值不存在于数组中，返回它将会被按顺序插入的位置。
     
    你可以假设数组中无重复元素。
     
    示例 1:
    输入: [1,3,5,6], 5
    输出: 2
     
    示例 2:
    输入: [1,3,5,6], 2
    输出: 1
     
    示例 3:
    输入: [1,3,5,6], 7
    输出: 4
     
    示例 4:
    输入: [1,3,5,6], 0
    输出: 0
     
    来源：力扣（LeetCode）
    链接：https://leetcode-cn.com/problems/search-insert-position

思路：二分查找
其实就是 bisect.bisect_left(nums, target)，但不敢真那么写。。

class Solution:
    def searchInsert(self, nums: List[int], target: int) -> int:
        lo, hi = 0, len(nums)
        while lo< hi:
            mid = (lo+hi)//2
            if nums[mid]>=target:
                hi = mid
            else:
                lo = mid +1
        return lo

34. 在排序数组中查找元素的第一个和最后一个位置
     
    给定一个按照升序排列的整数数组 nums，和一个目标值 target。找出给定目标值在数组中的开始位置和结束位置。
     
    你的算法时间复杂度必须是 O(log n) 级别。
     
    如果数组中不存在目标值，返回 [-1, -1]。
     
    示例 1:
     
    输入: nums = [5,7,7,8,8,10], target = 8
    输出: [3,4]
     
    示例 2:
     
    输入: nums = [5,7,7,8,8,10], target = 6
    输出: [-1,-1]
     
    来源：力扣（LeetCode）
    链接：https://leetcode-cn.com/problems/find-first-and-last-position-of-element-in-sorted-array

思路：二分查找
偷懒：python中使用bisect二分查找插入位置

class Solution:
    def searchRange(self, nums, target):
        lo = bisect.bisect_left(nums, target)
        hi = bisect.bisect(nums,target) - 1
        if lo > hi: return [-1,-1]
        return [lo, hi] 
        

不偷懒的话：

class Solution:
    def searchRange(self, nums, target):
        
        def helper(n):
            lo, hi = 0, len(nums)
            while lo< hi:
                mid = (lo+hi)//2
                if nums[mid] >= n:
                    hi = mid
                else:
                    lo = mid +1
            return lo

        lo = helper(target)
        hi = helper(target+1) -1
        if lo > hi: return [-1,-1]
        return [lo, hi]