【LeetCode每天一题】Search in Rotated Sorted Array II(在旋转数组中查找数字)

  Suppose an array sorted in ascending order is rotated at some pivot unknown to you beforehand. (i.e., [0,0,1,2,2,5,6] might become [2,5,6,0,0,1,2]). You are given a target value to search. If found in the array return true, otherwise return false.

Example 1:

Input: nums = [2,5,6,0,0,1,2], target = 0
Output: true

Example 2:

Input: nums = [2,5,6,0,0,1,2], target = 3
Output: false

Follow up:

  • This is a follow up problem to Search in Rotated Sorted Array, where nums may contain duplicates.
  • Would this affect the run-time complexity? How and why?

思路


       这道题在之前剑指offer中做过,两道题是一样的,都是利用二分查找的思想来进行查找,只不过这里的判断条件会复杂一些。因此我们可因此是旋转数组所以相当于会存在有序两个区间。每次进行查找时,我们先判断middle是否是target,不是时再判断nums[start] 与nums[middle]的大小关系。 然后再判断nums[start] ,nums[middle]与target的大小关系。从而可以缩小查找空间。另外因为数组中可能有重复的元素的情况,因此我们最初需要进行特殊判断是否存在,如果存在这种情况,我们只能使用顺序查找来得到结果。特殊的情况例如[0,0,0,0,0,0,-1,0,]。

图示思路


解决代码


 1 class Solution(object):
 2     def search(self, nums, target):
 3         """
 4         :type nums: List[int]
 5         :type target: int
 6         :rtype: bool
 7         """
 8         if not nums:       # 空列表直接返回
 9             return False
10         start, end = 0, len(nums)-1
11         middle = start +((end - start)>>1)       # 取中间值
12         if nums[start] == nums[middle] == nums[end]:       # 异常情况
13             for i in range(end+1):
14                 if nums[i] == target:
15                     return True
16             return False
17         
18         while start <= end:           # 二分查找
19             if nums[middle] == target:
20                 return True
21             if nums[start] <= nums[middle]:          # 说明再前半部分的有序序列中
22                 if nums[start] <= target and nums[middle] > target:     # 根据情况移动指针位置
23                     end = middle -1
24                 else:
25                     start = middle + 1
26             else:                                  #  否则再后半段的有序部分
27                 if nums[end] >= target and nums[middle] < target:      
28                     start = middle +1
29                 else:
30                     end = middle - 1
31             middle = start +((end - start)>>1)       # 重新取中间值
32         return False

猜你喜欢

转载自www.cnblogs.com/GoodRnne/p/10797074.html