LeetCode之HOT100--033 搜索旋转排序数组

95 阅读1分钟

这是我参与11月更文挑战的17天,活动详情查看:2021最后一次更文挑战」。

前言

一直都计划学习数据结构与基本算法,但是平时都看一阵停一阵。现在决心坚持下去,我准备从LeetCode的HOT100开始,每天完成1~2道习题,希望通过这种方式养成持续学习的习惯。因为我是做iOS开发的,主要是用Objective-C语言,最近也在学习Swift,所以本系列的题解都将使用swift语言完成,本文更新的是LeetCode中HOT100的第17题033 搜索旋转排序数组。

题目

整数数组 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,5,6,7] 在下标 3 处经旋转后可能变为 [4,5,6,7,0,1,2] 。
给你 旋转后 的数组 nums 和一个整数 target ,如果 nums 中存在这个目标值 target ,则返回它的下标,否则返回 -1 。
示例 1:

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

示例 2:

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

示例 3:

输入:nums = [1], target = 0
输出:-1

提示:

1. 1 <= nums.length <= 5000
2. -10^4 <= nums[i] <= 10^4
3. nums 中的每个值都 独一无二
4. 题目数据保证 nums 在预先未知的某个下标上进行了旋转
5. -10^4 <= target <= 10^4

分析

对于有序数组,可以使用二分查找的方法查找元素。但是这道题中,数组本身不是有序的,进行旋转后只保证了数组的局部是有序的,这还能进行二分查找吗?答案是可以的。
可以发现的是,我们将数组从中间分开成左右两部分的时候,一定有一部分的数组是有序的。拿示例来看,我们从 6 这个位置分开以后数组变成了 [4, 5, 6] 和 [7, 0, 1, 2] 两个部分,其中左边 [4, 5, 6] 这个部分的数组是有序的,其他也是如此。
这启示我们可以在常规二分查找的时候查看当前 mid 为分割位置分割出来的两个部分 [left, mid] 和 [mid + 1, right] 哪个部分是有序的,并根据有序的那个部分确定我们该如何改变二分查找的上下界,因为我们能够根据有序的那部分判断出 target 在不在这个部分:\

  1. 如果 [left, mid - 1] 是有序数组,且 target 的大小满足 [nums[left],nums[mid]),则我们应该将搜索范围缩小至 [left, mid - 1],否则在 [mid + 1, right] 中寻找。
  2. 如果 [mid, right] 是有序数组,且 target 的大小满足(nums[mid+1],nums[right]],则我们应该将搜索范围缩小至 [mid + 1, right],否则在 [left, mid - 1] 中寻找。

33_fig1.png

需要注意的是,二分的写法有很多种,所以在判断 target 大小与有序部分的关系的时候可能会出现细节上的差别。

题解

class KLLC033 {

    func search(_ nums: [Int], _ target: Int) -> Int {
        if nums.isEmpty {
            return -1
        }
        //left right的初始化
        var left = 0, right = nums.count - 1
        while left <= right {
            //取 mid
            let mid = (left + right) >> 1
            //mid为目标值 则返回 over
            if nums[mid] == target {
                return mid
            }
            //[left, mid]为有序数组的条件是 nums[left] <= nums[mid]
            if nums[left] <= nums[mid] {
                //target 的大小满足 [nums[left],nums[mid])
                if nums[left] <= target && nums[mid] > target {
                    right = mid - 1
                } else {
                   left = mid + 1
                }
            } else {
            //否则是 [mid, right]为有序数组
                //target 的大小满足 [nums[],nums[right])
                if nums[mid] < target && nums[right] >= target {
                    left = mid + 1
                } else {
                   right = mid - 1
                }
            }
        }
        //没有找到则返回 -1
        return -1
    }
    

    

}