day04-[34] 在排序数组中查找元素的第一个和最后一个位置(二分查找)

191 阅读1分钟

[34] 在排序数组中查找元素的第一个和最后一个位置

描述

给你一个按照非递减顺序排列的整数数组 nums,和一个目标值 target。请你找出给定目标值在数组中的开始位置和结束位置。

如果数组中不存在目标值 target,返回 [-1, -1]

你必须设计并实现时间复杂度为 O(log n) 的算法解决此问题。 示例 1:

输入: nums = [5,7,7,8,8,10], target = 8
输出: [3,4]

示例 2:

输入: nums = [5,7,7,8,8,10], target = 6
输出: [-1,-1]

示例 3:

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

提示:

  • 0 <= nums.length <= 105
  • -109 <= nums[i] <= 109
  • nums 是一个非递减数组
  • -109 <= target <= 109

解题方案

Swift代码如下:

func searchRange(_ nums: [Int], _ target: Int) -> [Int] {

    var rightBorder = -2
    var leftBoder = -2 
    leftBoder = funcleftBorder(nums, target)
    rightBorder = funcrightBorder(nums,target)
    if(leftBoder == -2 || rightBorder == -2){
        return [-1,-1]
    }
    if(rightBorder - leftBoder > 1){
        return [leftBoder+1,rightBorder-1]
    }
    return [-1,-1]
  
    }
    //寻找左边界
    func funcleftBorder(_ nums:[Int],_ target:Int) -> Int {
        var left = 0
        var right = nums.count - 1 
        while left <= right {
            var midle = left + (right - left)/2
            if target > nums[midle] {
                left = midle + 1
            }else {
                right = midle - 1   
            }
        }
        return right
    }
    //寻找右边界
    func funcrightBorder(_ nums:[Int],_ target:Int) -> Int {
           var left = 0
           var right = nums.count - 1
           while left <= right {
               var midle = left + (right - left)/2
               if target < nums[midle] {
                   right = midle - 1
               }else {
                   left = midle + 1   
               }
           }
           return left
       }
    }