[Leetcode][python]搜索旋转排序数组/搜索旋转排序数组 II

Search in Rotated Sorted Array

题目大意

把一个严格升序的数组进行旋转,如[0,1,2,3,4,5]旋转3位成为[3,4,5,0,1,2]。在这样的数组中找到目标数字。如果存在返回下标,不存在返回-1。
输入: nums = [4, 5, 6, 7, 0, 1, 2], target = 6 输出: 2
输入: nums = [4, 5, 6, 7, 0, 1, 2], target = 3 输出: -1

解题思路

二分搜索是针对有序数组而言,对于中间有次转折的有序数组,只是要多区分几种情况,二分搜索依然是适用的。
主要就是判断mid的左右哪边是中断的

代码

该解法的判断表达是对称的结构,有助于记忆

class Solution(object):
    def search(self, nums, target):
        """
        :type nums: List[int]
        :type target: int
        :rtype: int
        """
        left = 0; right = len(nums) - 1
        while left <= right:
            mid = (left + right) / 2
            if target == nums[mid]:
                return mid
            if nums[mid] >= nums[left]:  # 中点大于最左边,说明左边区间没有混乱
                if target < nums[mid] and target >= nums[left]:  # 此时若目标数小于中间数而且大于最左边,说明就在左边区间
                    right = mid - 1
                else:  # 不然就在右边区间
                    left = mid + 1
            else:  # 中点小于最右边,说明右边区间没有混乱
                if target > nums[mid] and target <= nums[right]:  # 此时若目标数大于中间数而且小于最右边,说明就在右边区间
                    left = mid + 1
                else:  # 不然就在左边区间
                    right = mid - 1
        return -1  # 找不到

Search in Rotated Sorted Array II

题目大意

把一个有重复的排序数组进行旋转,如[0,1,1,1,2,3,4,5]旋转3位成为[3,4,5,0,1,1,1,2]。在这样的数组中判断目标数字是否存在。

解题思路

和上题其实唯一的区别在于如果重复的数字中间进行了旋转,所以只要加入

if nums[left] == nums[mid] == nums[right]:
    left += 1
    right -= 1

其他相同。

代码

class Solution(object):
    def search(self, nums, target):
        """
        :type nums: List[int]
        :type target: int
        :rtype: bool
        """
        left = 0; right = len(nums)-1
        while left <= right:
            mid = (left + right) // 2
            if nums[mid] == target: 
                return True
            if nums[left] == nums[mid] == nums[right]:  # 所有指针都一个数字,说明旋转时有的在左边有的在右边,需要往中间看
                left += 1
                right -= 1
            elif nums[mid] >= nums[left]:  # 中点大于最左边,说明左边区间没有混乱
                if target < nums[mid] and target >= nums[left]:  # 此时若目标数小于中间数而且大于最左边,说明就在左边区间
                    right = mid - 1
                else:  # 不然就在右边区间
                    left = mid + 1
            else:  # 中点小于最右边,说明右边区间没有混乱
                if target > nums[mid] and target <= nums[right]:  # 此时若目标数大于中间数而且小于最右边,说明就在右边区间
                    left = mid + 1
                else:  # 不然就在左边区间
                    right = mid - 1
        return False

总结

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值