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

    科技2023-09-29  80

    题目

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

    ( 例如,数组 [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 可能包含重复元素。这会影响到程序的时间复杂度吗?会有怎样的影响,为什么?

    思路

    同 搜索旋转排序数组 一样,我们还是采用二分搜索,显然,在我们第一次进行二分的时候,切分的两段数组必然有一个是有序的,在不包含重复元素的情况下,判断这两段数组哪个是有序的是很容易的,我们只需分别比较两段数组的头部和尾部即可,但是在又重复元素的情况下,例如 1,3,1,1,1,切分后得到两段 1,3,1 和 1,1,1 ,我们无法通过比较头部和尾部来判断是否有序,所以一个简单的做法是,将 mid 指针一分为二,mid1 向左移,直到左边这一段数组有序,mid2 向右移,直到右边这个一段数组有序,这样得到的两段数组都是有序的了,接下来直接二分即可;

    代码

    class Solution { public boolean search(int[] nums, int target) { int left = 0, right = nums.length-1; while (left <= right){ int mid = (left+right)/2; if (nums[mid] == target) return true; int mid1 = mid, mid2 = mid; while (mid1 > left && nums[mid1] <= nums[left]) { mid1--; if (nums[mid1] == target) // 避免错过正确解 return true; } while (mid2 < right && nums[mid2] >= nums[right]) { mid2++; if (nums[mid2] == target) // 避免错过正确解 return true; } if (target <= nums[mid1] && target >= nums[left]) right = mid1-1; else if (target >= nums[mid2] && target <= nums[right]) left = mid2+1; else break; } return false; } }
    Processed: 0.019, SQL: 8