一、题目描述
整数数组 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
。
你必须设计一个时间复杂度为 O(log n)
的算法解决此问题。
示例 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
-104 <= nums[i] <= 104
nums 中的每个值都 独一无二
题目数据保证 nums 在预先未知的某个下标上进行了旋转
-104 <= target <= 104
二、代码
代码如下:
class Solution:
def search(self, nums: List[int], target: int) -> int:
#z表示左坐标,y表示右坐标
z = 0
y = len(nums) - 1
index = -1
while z <= y:
m = int((z + y) / 2)
if target == nums[m]:
print("m:",m)
#如果此时的目标值等于当前的中间值,则找到
return m
#如果左边部分有序:
if nums[0] <= nums[m]:
#如果目标值在左边部分范围
if nums[0] <= target < nums[m]:
#缩小左边部分的右边界
y = m - 1
#如果目标值超出了左边部分范围
else:
#跳转至右边部分
z = m + 1
#如果右边部分有序(至少有一边是有序的):
else:
#如果目标值在右边部分范围内
if nums[m] < target <= nums[len(nums) - 1]:
#缩小右边部分的左边界
z = m + 1
#如果目标值超出了右边部分范围
else:
#跳转至左边界
y = m - 1
print(-1)
#如果最后仍然没有找到目标值,则返回-1
return -1
三、解题思路
PS:个人能力有限,不考虑时间空间最优。
本题题意就是在数组中找对应元素的下标,只不过要求使用时间复杂度为O(logn)的方法,并且给出的初始数组nums还是“旋转过后”的。
旋转过后的数组nums仍然具有一些特征:nums中可以看出是左右两部分,其中每一部分都是升序的,但两部分的交界处不符合升序。
为了使解法的时间复杂度为O(logn),则可以采用二分法进行查找,只不过此时的二分法相对于来说有些不同,需要进行判断:取中点位置m作为中间坐标,将nums分成2部分,则此时这两部分中一定存在一部分是完全升序的,另一部分则可能是也可能不是。
本题的思路就是将那一部分有序的数组进行二分查找,如果target不在该范围内,就重新设置左、右范围坐标,重新查找有序的那一部分,直至左坐标大于右坐标或者target等于中间坐标对应的nums值。