LeetCode 33 35 搜索插入位置 搜索旋转排序数组 二分法

写在前面

将两道题放在一起的原因是两道题比较相似，35是简单二分，33是特殊情况的二分。

题目 33 搜索插入位置

给定一个排序数组和一个目标值，在数组中找到目标值，并返回其索引。如果目标值不存在于数组中，返回它将会被按顺序插入的位置。

你可以假设数组中无重复元素。

示例 1:

输入: [1,3,5,6], 5
输出: 2
示例 2:

输入: [1,3,5,6], 2
输出: 1
示例 3:

输入: [1,3,5,6], 7
输出: 4
示例 4:

输入: [1,3,5,6], 0
输出: 0

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/search-insert-position
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

思路及代码

简单二分法

fun searchInsert(nums: IntArray, target: Int): Int {
    var left = 0
    var right = nums.size-1
    var mid = 0
    while (left<=right){
        mid = left+(right-left)/2
        if (nums[mid] < target)
            left = mid + 1
        else if(nums[mid]>target)
            right = mid-1
        else
            return mid
    }
    return left
}

题目 33 搜索旋转排序数组

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

( 例如，数组 [0,1,2,4,5,6,7] 可能变为 [4,5,6,7,0,1,2] )。

搜索一个给定的目标值，如果数组中存在这个目标值，则返回它的索引，否则返回 -1 。

你可以假设数组中不存在重复的元素。

你的算法时间复杂度必须是 O(log n) 级别。

示例 1:

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

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/search-in-rotated-sorted-array
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

思路及代码

第一个解法，判定当前数的范围，在left…mid之间还是mid…right之间，分了两个方向进行二分。代码演示第一种解法。
第二个解法，使用二分查找出旋转的点，然后判定target在旋转点的哪一边，对target存在的一边进行第二次二分。

fun search(nums: IntArray, target: Int): Int {
    if(nums.isEmpty()) return -1
    var left = 0
    var right = nums.size-1
    while (left<right){
        var mid = left+(right-1)
        if(nums[mid]<nums[right]){
            if(target in nums[mid]+1..nums[right]){
                left = mid+1
            }else{
                right = mid
            }
        }else{
            if(target in nums[left]..nums[mid]){
                right = mid
            }else{
                left = mid+1
            }
        }
    }

    return if(nums[left]!=target) -1 else left
}