【LeetCode-查找】搜索旋转排序数组 II

题目描述

假设按照升序排序的数组在预先未知的某个点上进行了旋转。
( 例如,数组 [0,0,1,2,2,5,6] 可能变为 [2,5,6,0,0,1,2] )。
编写一个函数来判断给定的目标值是否存在于数组中。若存在返回 true,否则返回 false。
示例:

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

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

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

思路

这题是搜索旋转排序数组的延伸版。在“搜索旋转数组”那题中,首先通过num[left],nums[right]和nums[mid]之间的关系来判断接下来搜索的范围。在这题中,数组中可能存在重复元素,也就是可能存在nums[left]==nums[mid]==nums[right]的情况,例如[1,3,1,1,1]、[1,1,1,3,1]。这种情况出现时,我们没有办法缩小搜索的范围,只能在[left, right]范围内顺序查找。所以,代码就是在搜索旋转排序数组这题代码的基础上加了一个判断条件,具体如下:

class Solution {
public:
    bool search(vector<int>& nums, int target) {
        if(nums.empty()) return false;

        int left = 0, right = nums.size()-1;
        while(left<=right){
            int mid = left+(right-left)/2;
            if(nums[mid]==target) return true;
            if(nums[left]==nums[mid] && nums[mid]==nums[right]){    // 三者相等,遍历寻找
                for(int i=left; i<=right; i++){
                    if(nums[i]==target) return true;
                }
                return false;
            }
            if(nums[left]<=nums[mid]){
                if(nums[left]<=target && target<nums[mid]) right = mid-1;
                else left = mid+1;
            }else{    // 也可以写成else if(nums[mid]<=nums[right]),更易读
                if(nums[mid]<target && target<=nums[right]) left = mid+1;
                else right = mid-1;
            }
        }
        return false;
    }
};
  • 时间复杂度:O(n)+O(logn)
    这个时间复杂度不一定准确。
  • 空间复杂度:O(1)

猜你喜欢

转载自www.cnblogs.com/flix/p/12803747.html
今日推荐