33. 搜索旋转排序数组

整数数组 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
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值