代码随想录算法训练营第一天| 704. 二分查找，35.搜索插入位置 和 34. 在排序数组中查找元素的第一个和最后一个位置

题目建议

先把 704写熟练，要熟悉 根据 左闭右开，左闭右闭 两种区间规则 写出来的二分法。

题目链接：. - 力扣（LeetCode）

文章讲解：代码随想录

视频讲解：手把手带你撕出正确的二分法 | 二分查找法 | 二分搜索法 | LeetCode：704. 二分查找_哔哩哔哩_bilibili

704 二分查找

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

示例 1:

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

示例 2:

输入: nums = [-1,0,3,5,9,12], target = 2
输出: -1
解释: 2 不存在 nums 中因此返回 -1

先讲讲我犯的错，开始没怎么理解二分法，以为首尾相减除以二就得到了中间的索引，结果推算的时候发现不对，得加上左指针left；

其次，我是用Python刷的题，除以二的时候很有可能除不了，c++里面直接用/能够直接取整，python里面得用//

然后，关于left = 0，right是取len（nums）还是取len(nums) - 1，while循环里面是left < right还是left<=right，这个都需要自己推算一下，

1、left = 0，right =len(nums),while(left<right)缺少一些情况

2、left = 0，right =len(nums),while(left<=right) 索引会溢出

3、left = 0，right =len(nums) - 1,while(left<right)部分测试用例能通过，例如当nums = [5]，只包含一个元素时，无法进入循环

代码实现：

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

35.搜索插入位置

给定一个排序数组和一个目标值，在数组中找到目标值，并返回其索引。如果目标值不存在于数组中，返回它将会被按顺序插入的位置。

请必须使用时间复杂度为 O(log n) 的算法。

示例 1:

输入: nums = [1,3,5,6], target = 5
输出: 2

示例 2:

输入: nums = [1,3,5,6], target = 2
输出: 1

示例 3:

输入: nums = [1,3,5,6], target = 7
输出: 4

二分法的变种题目，整体的代码和上一题差不多，就是要考虑到如果原来数组中没有该目标值，需要在哪个位置插入的问题。可以注意到，停止循环的条件是left>right或者找到target的索引，所以没找到target时，left是大于right的，且left = right+1，此时中间的都遍历结束了，所以只需要在left的位置插入target，也就是最后返回left即可

代码

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

34.在排序数组中查找元素的第一个和最后一个位置

给你一个按照非递减顺序排列的整数数组 nums，和一个目标值 target。请你找出给定目标值在数组中的开始位置和结束位置。

如果数组中不存在目标值 target，返回 [-1, -1]。

你必须设计并实现时间复杂度为 O(log n) 的算法解决此问题。

示例 1：

输入：nums = [5,7,7,8,8,10], target = 8
输出：[3,4]

示例 2：

输入：nums = [5,7,7,8,8,10], target = 6
输出：[-1,-1]

示例 3：

输入：nums = [], target = 0
输出：[-1,-1]

这个题我一开始没用二分做，想着可以先找到第一个索引，用指针记录，然后从后往前遍历，找到最后的索引，最后用数组输出，代码先贴在这里

代码

class Solution:
    def searchRange(self, nums: List[int], target: int) -> List[int]:
        arr = []
        l = len(nums) - 1
        i = 0
        while i <= l:
            if nums[i] == target:
                arr.append(i)
                while l >= i:
                    if nums[l] == target:
                        arr.append(l)
                        return arr
                    else:
                        l -= 1
            i += 1
        return [-1, -1]

后来用二分法重新做了一遍，用了两次二分法，找到fisrt和last索引

class Solution:
    def searchRange(self, nums: List[int], target: int) -> List[int]:
        arr = []
        left = 0
        right = len(nums) - 1
        first = -1
        last = -1
        while left <= right:
            mid = (right - left) // 2 + left
            if nums[mid] > target:
                right = mid - 1
            elif nums[mid] < target:
                left = mid + 1
            else:
                first = mid
                right = mid - 1  #重要
        arr.append(first)
        left = 0
        right = len(nums) - 1
        while left <= right:
            mid = (right - left) // 2 + left
            if nums[mid] > target:
                right = mid - 1
            elif nums[mid] < target:
                left = mid + 1
            else:
                last = mid
                left = mid + 1 #重要
        arr.append(last)
        return arr

第一次二分查找，在找到target之后，用fisrt记录下来，为了保证是第一个索引位置，需要通过right =mid-1来缩小范围

第二次二分查找，在找到target之后，用last记录下来，为了保证是最后一个索引位置，需要通过left = mid + 1来缩小右边的范围