81. 搜索旋转排序数组 II(python3)

题目:https://leetcode-cn.com/problems/search-in-rotated-sorted-array-ii/

假设按照升序排序的数组在预先未知的某个点上进行了旋转。

( 例如,数组 [0,0,1,2,2,5,6] 可能变为 [2,5,6,0,0,1,2] )。

编写一个函数来判断给定的目标值是否存在于数组中。若存在返回 true,否则返回 false。

示例 1:

输入: nums = [2,5,6,0,0,1,2], target = 0
输出: true
示例 2:

输入: nums = [2,5,6,0,0,1,2], target = 3
输出: false
进阶:

这是 搜索旋转排序数组 的延伸题目,本题中的 nums  可能包含重复元素。
这会影响到程序的时间复杂度吗?会有怎样的影响,为什么?

 

解题思路:

二分查找,因数列进行了一次旋转,所以在二分点mid = start + (end - start) // 2的左右两边一定有一边是有序的,只需比较target与有序边的边界值判断是否在该区间。
此题需要注意的是,序列存在相同值,所以会存在[1,1,3,1]这种情况,第一次mid指向了下标1,由于nums[start] == nums[mid] == nums[end],无法判断两边哪边有序,此时需要舍弃相同的值,即当start位与mid位相同时,start++,end位与mid位相同时,end--。

代码:

class Solution:
    def search(self, nums: List[int], target: int) -> bool:
        start = 0
        end = len(nums) - 1
        
        while(start <= end):
            mid = start + (end - start) // 2

            if(nums[mid] == target):
                    return True

            if(nums[mid] < nums[end]):
                if(nums[mid] <= target <= nums[end]):
                    start = mid + 1
                else:
                    end = mid - 1
            elif(nums[start] < nums[mid]):
                if(nums[start] <= target <= nums[mid]):
                    end = mid - 1
                else:
                    start = mid + 1
            elif(nums[start] == nums[mid]):
                start = start + 1
            elif(nums[end] == nums[mid]):
                end = end - 1
                
        
        return False

 

测试用例:

if __name__=="__main__":
    nums = [2,5,6,0,0,1,2]
    target = 0
    res = Solution().search(nums, target)
    print(res)

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值