搜索旋转排序数组 II

搜索旋转排序数组 II

CategoryDifficultyLikesDislikes
algorithmsMedium (40.98%)717-
Tags

array | binary-search

Companies

Unknown

已知存在一个按非降序排列的整数数组 nums ,数组中的值不必互不相同。

在传递给函数之前,nums 在预先未知的某个下标 k0 <= 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:

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

示例 2:

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

提示:

  • 1 <= nums.length <= 5000
  • -10<sup>4</sup> <= nums[i] <= 10<sup>4</sup>
  • 题目数据保证 nums 在预先未知的某个下标上进行了旋转
  • -10<sup>4</sup> <= target <= 10<sup>4</sup>

进阶:

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

Discussion | Solution

解法

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
// @lc code=start
impl Solution {
    /// ## 解题思路
    /// - 二分查找
    /// 1. 在<<33.搜索旋转排序数组>>题目基础上, 增加对重复数字的判断;
    pub fn search(nums: Vec<i32>, target: i32) -> bool {
        if nums.len() == 1 {
            return nums[0] == target;
        }
        let (mut l, mut r) = (0, nums.len() - 1);
        while l < r {
            let m = (l + r + 1) / 2;
            if target == nums[l] || target == nums[r] || target == nums[m]  {
                return true;
            }
            if nums[m] == nums[l] {
                l += 1;
            } else if nums[m] == nums[r] {
                r -= 1;
            } else if nums[m] > nums[l] {
                if target > nums[l] && target < nums[m] {
                    r = m - 1;
                } else {
                    l = m + 1;
                }
            } else {
                if target > nums[m] && target < nums[r] {
                    l = m + 1;
                } else {
                    r = m - 1;
                }
            }
        }

        return false;
    }
}
// @lc code=end

struct Solution;

updatedupdated2024-08-252024-08-25