LeetCode 81 Search in Rotated Sorted Array II (Tag:Array Difficulty:Medium)

这是我参与11月更文挑战的第8天，活动详情查看：2021最后一次更文挑战

前言

关于 LeetCode 数组类型题目的相关解法，可见LeetCode 数组类型题目做前必看，分类别解法总结了题目，可以用来单项提高。觉得有帮助的话，记得多多点赞关注哦，感谢！

题目描述

已知存在一个按非降序排列的整数数组 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,4,4,5,6,6,7] 在下标 5 处经旋转后可能变为 [4,5,6,6,7,0,1,2,4,4] 。

给你 旋转后 的数组 nums 和一个整数 target ，请你编写一个函数来判断给定的目标值是否存在于数组中。如果 nums 中存在这个目标值 target ，则返回 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

链接：leetcode-cn.com/problems/se…

题解

这道题目和 leetcode 33 都是在旋转数组中找 target 元素，区别在于 33 题是没有重复元素的有序数组，本题是存在重复元素的有序数组。所以在 33 题解法的基础上，多了一个分支。

在 33 题解法中，分为 nums[mid] < nums[r] 和 nums[mid] > nums[r] 两个分支，这道题目多了一个 nums[mid] === nums[r] 的分支，其他两个分支处理方法和 33 题是一致的，主要是通过判断 target 落在哪个区间来调整 l 和 r 的大小。对与 nums[mid] === nums[r] 分支来讲，其实只需要缩小 r 的范围就好了，即 r--。

具体代码见下方，时间复杂度为 O(logn)

/**
 * @param {number[]} nums
 * @param {number} target
 * @return {boolean}
 */
var search = function(nums, target) {
    let l = 0, r = nums.length - 1
    
    while(l <= r) {
        const mid = l + Math.floor((r - l) / 2)
        if (nums[mid] === target) return true
        if (nums[mid] < nums[r]) {
            if (nums[mid] < target && target <= nums[r]) {
                l = mid + 1
            } else {
                r = mid - 1
            }
        } else if (nums[mid] > nums[r]) {
            if (nums[mid] > target && target >= nums[l]) {
                r = mid - 1
            } else {
                l = mid + 1
            }
        } else {
            r--
        }
    } 
    return false
}