二分法真的是所谓的玄学代码,反正我昨天刚刚学过的,今天遇到一个题,条件又不会判断了,而且我昨天写的博客不见了,气哭。
昨天看到一个很好的讲解二分法的大佬。
其实主要就是一些细节的问题,能够返回正确的值,并且能够不造成死循环。
下面是一道leetcode上的问题:
假设按照升序排序的数组在预先未知的某个点上进行了旋转。
( 例如,数组 [0,1,2,4,5,6,7] 可能变为 [4,5,6,7,0,1,2] )。
搜索一个给定的目标值,如果数组中存在这个目标值,则返回它的索引,否则返回 -1 。
你可以假设数组中不存在重复的元素。
你的算法时间复杂度必须是 O(log n) 级别。
class Solution:
def search(self, nums: List[int], target: int) -> int:
if len(nums)==0:
return -1
low, high = 0, len(nums)-1
while low<=high:
mid = (low +high)//2
if nums[mid]==target:
return mid
elif nums[mid]>=nums[low]: #必须有=
if target>=nums[low] and target<nums[mid]:
high = mid-1
else:
low = mid+1
elif nums[mid]<nums[low]:
if target>nums[mid] and target<=nums[high]:
low = mid+1
else:
high=mid-1
return -1
算法思路是这样的:
由于题目说数字了无重复,举个例子:
1 2 3 4 5 6 7 可以大致分为两类,
第一类 2 3 4 5 6 7 1 这种,也就是 nums[start] <= nums[mid]。此例子中就是 2 <= 5。
这种情况下,前半部分有序。因此如果 nums[start] <=target<nums[mid],则在前半部分找,否则去后半部分找。
第二类 6 7 1 2 3 4 5 这种,也就是 nums[start] > nums[mid]。此例子中就是 6 > 2。
这种情况下,后半部分有序。因此如果 nums[mid] <target<=nums[end],则在后半部分找,否则去前半部分找。
(来源leetcode)
这里有几个问题需要注意:
1、一开始总是出现超出时间限制的错误,这种多半是出现了死循环,原因是我标记处少写了等于号,之前不太明白,数组里明明没有相等的值,有可能会出现等于的情况么?可是想一想,玩意low和mid指向的是同一位置呢,比如说[2,1]这个数组,mid和low都指向2,如果不对这种情况进行处理,会造成死循环
2、内部循环也不能忘记=,这里我表示的是target在数组的前半部分,当然可以target=nums[mid].