N久N久没有刷过题目了,昨天做了一道标签为“简单”的二分类题目,写的一塌糊涂。。。
看到一篇很棒的二分法总结,我接着跟一跟这篇总结继续刷一下二分法相关的题目。
写对二分查找不能靠模板,需要理解加练习 (附练习题,持续更新) - 搜索插入位置 - 力扣(LeetCode) (leetcode-cn.com)
702题:二分查找
给定一个 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]之间。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/binary-search
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
- 第一次错误的提交
class Solution:
def search(self, nums: List[int], target: int) -> int:
left, right = 0, len(nums) -1
while(left < right): # 区别只在这里
mid = left + (right - left) // 2 # 注意,这里的写法,可以防止left+right的结果溢出
if target < nums[mid]:
right = mid - 1
elif target > nums[mid]:
left = mid + 1
else:
return mid
return -1
- 正确的提交
区别是"while(left <= right): # 区别只在这里",如果是"<",则忽略了列表只有一个元素的情况
class Solution:
def search(self, nums: List[int], target: int) -> int:
left, right = 0, len(nums) -1
while(left <= right): # 区别只在这里
mid = left + (right - left) // 2
if target < nums[mid]:
right = mid - 1
elif target > nums[mid]:
left = mid + 1
else:
return mid
return -1