整数数组 nums 按升序排列,数组中的值 互不相同 。
在传递给函数之前,nums 在预先未知的某个下标 k(0 <= k < nums.length)上进行了 旋转,使数组变为 [nums[k], nums[k+1], …, nums[n-1], nums[0], nums[1], …, nums[k-1]](下标 从 0 开始 计数)。例如, [0,1,2,4,5,6,7] 在下标 3 处经旋转后可能变为 [4,5,6,7,0,1,2] 。
给你 旋转后 的数组 nums 和一个整数 target ,如果 nums 中存在这个目标值 target ,则返回它的索引,否则返回 -1 。
示例 1:
输入:nums = [4,5,6,7,0,1,2], target = 0
输出:4
示例 2:
输入:nums = [4,5,6,7,0,1,2], target = 3
输出:-1
示例 3:
输入:nums = [1], target = 0
输出:-1
提示:
1 <= nums.length <= 5000
-10^4 <= nums[i] <= 10^4
nums 中的每个值都 独一无二
nums 肯定会在某个点上旋转
-10^4 <= target <= 10^4
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/search-in-rotated-sorted-array
有几个想法,首先是暴力法,全部遍历,但是也有改进的地方,因为nums[0]是左边有序数列的最小值,nums[-1]是右边有序数列的最大值,那么进行一个初步判断然后进行迭代。
class Solution:
def search(self, nums: List[int], target: int) -> int:
if target not in nums:
return -1
index = 0
if target > nums[0]:
while 1:
if target == nums[index]:
return index
index += 1
else:
index = -1
while 1:
if target == nums[index]:
return len(nums) + index
index -= 1
第二种是还原法,将被旋转的数组先还原成有序的,记录下旋转的地方。
class Solution:
def search(self, nums: List[int], target: int) -> int:
# 还原法
movement = 0
while nums[0] > nums[-1]:
nums = [nums.pop()] + nums
movement += 1
l, r = 0, len(nums) - 1
while l < r:
mid = int((l + r) / 2)
if nums[mid] > target:
r = mid
elif nums[mid] < target:
l = mid + 1
else:
if mid < movement:
return len(nums) - movement + mid
else:
return mid - movement
if nums[l] == target:
return len(nums) - movement + l if l < movement else l - movement
return -1
最后一个方法就是二分法。
class Solution:
def search(self, nums: List[int], target: int) -> int:
# 二分法
def helper_(l, r):
while l < r:
mid = int((l+r)/2)
if nums[mid] < target:
l = mid+1
elif nums[mid] > target:
r = mid
else:
return mid
if nums[l] == target:
return l
return -1
l, r = 0, len(nums)-1
while l < r:
mid = int((l+r)/2)
# 说明右半部分是有序的(从小到大)
if nums[mid] < nums[r]:
if nums[mid] < target < nums[r]:
return helper_(mid, r)
elif target == nums[mid]:
return mid
elif target == nums[r]:
return r
r = mid
# 说明左半部分是有序的
else:
if nums[l] < target < nums[mid]:
return helper_(l, mid)
elif target == nums[mid]:
return mid
elif target == nums[l]:
return l
l = mid + 1
if nums[l] == target:
return l
return -1
写的有点复杂,简化一下:
class Solution:
def search(self, nums: List[int], target: int) -> int:
left, right = 0, len(nums) - 1
while left <= right:
mid = int((left + right) / 2)
if nums[mid] == target:
return mid
# 说明右半部分有序
if nums[mid] < nums[right]:
if nums[mid] <= target <= nums[right]:
left = mid+1
else:
right = mid-1
# 说明左半部分有序
else:
if nums[left] <= target <= nums[mid]:
right = mid-1
else:
left = mid+1
return -1