day01-二分查找

一、二分查找

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

来源：力扣（LeetCode）

解题思路：
1.设置左右指针
2.找到中间位置并判断该位置值是否等于 target
3.nums[middle] == target 则返回该位置下标
4.nums[middle] > target 则右侧指针移到中间
5.nums[middle] < target 则左侧指针移到中间
6.如果循环中没有找到target，则退出循环，返回-1

# @-*- coding: utf-8 -*-
# @Time: 2022/3/26 09:30
# @File: day01-二分查找.py
# @Author: 师玉幺叔
# @Software: Pycharm

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


if __name__ == '__main__':
    nums = [-1, 0, 3, 5, 9, 12]
    target = 9
    re = Solution()
    res = re.search(nums, target)
    print(res)

二、插入搜索位置

给定一个排序数组和一个目标值，在数组中找到目标值，并返回其索引。如果目标值不存在于数组中，返回它将会被按顺序插入的位置。
请必须使用时间复杂度为 O(log n) 的算法。

来源：力扣 (LeetCode)

解题思路：
1.设置左右指针
2.找到中间位置并判断该位置值是否等于 target
3.nums[middle] == target 则返回该位置下标
4.nums[middle] > target 则右侧指针移到中间
5.nums[middle] < target 则左侧指针移到中间
6.如果循环中没有找到target，则退出循环，返回左指针

示例：
nums = [1, 3, 5, 6]
target = 7
left = 0, right = len(nums) - 1 -----> 3
以 left <= right 作为循环条件

第一次循环
中间位置为：middle = (left + right) // 2 —> (0+3)//2-------->1
nums[middle]---->nums[1]---->3
target: 7 > 3
left=middle+1----->1+1----->2
right----->3

第二次循环
中间位置为：middle = (left + right) // 2 —> (2+3)//2----->2
nums[middle]---->nums[2]---->5
target: 7 > 5
left=middle+1------>2+1------->3
right------>3

第三次循环
中间位置为：middle = (left + right) // 2 —> (3+3)//2------>3
nums[middle]---->nums[3]---->6
target: 7 > 6
left=middle+1----->3+1----->4
right---->3

此时 left > right 退出循环
left即为target的插入位置--------> 4

# @-*- coding: utf-8 -*-
# @Time: 2022/3/26 10:08
# @File: day01-搜索插入位置.py
# @Author: 师玉幺叔
# @Software: Pycharm

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


if __name__ == '__main__':
    nums = [1, 3, 5, 6]
    target = -1
    re = Solution().searchInsert(nums, target)
    print(re)