题目
给定一个 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
提示:
你可以假设 nums 中的所有元素是不重复的。
n 将在 [1, 10000]之间。
nums 的每个元素都将在 [-9999, 9999]之间。
解题思路
最基础的二分查找,每次丢弃的点包含当前middle指向的数
成绩
时间>72.76%
空间>6.67%
代码
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
变形
如果寻找的数组中有重复的数字:
- 需要找到第一个出现的数字。第一个其实就是最左边出现的数字,在以上的基础上修改思路,当找到target后,继续在[left, middle]中找。代码变为:当
nums[middle] == target
时,right = middle
,同时判断条件变为while left < right
。退出循环后需要判断right指向的数字是否为target
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:
right = middle
if nums[right] == target:
return right
return -1
- 需要找到最后一个出现的数字。其实就是找到符合的数字的最后一个位置,和上面情况类似,不过是right和left互换。需要注意的细节在于:
- 由于普通情况下middle是向下取整(靠近左边)的,所以
left = middle
会造成循环问题,改变middle为向上取整即可 - 注意判断越界
- 由于普通情况下middle是向下取整(靠近左边)的,所以
class Solution:
def search(self, nums: List[int], target: int) -> int:
left, right = 0, len(nums) - 1
while left < right:
middle = (left + right + 1) // 2
if target > nums[middle]:
left = middle + 1
elif target < nums[middle]:
right = middle - 1
else:
left = middle
if left < len(nums) and nums[left] == target:
return left
return -1