leetcode刷题笔记三十四 在排序数组中查找元素的第一个和最后一个位置

leetcode刷题笔记三十四 在排序数组中查找元素的第一个和最后一个位置

源地址:34. 在排序数组中查找元素的第一个和最后一个位置

问题描述:

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

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

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

示例 1:

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

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

代码补充:

//本题较为简单,通过二分法查找到目标值,然后向两侧扩展,寻找左右边界即可
object Solution {
    def searchRange(nums: Array[Int], target: Int): Array[Int] = {
        if (nums.length == 0) return Array(-1,-1)
        if (nums.length == 1 && nums(0) == target) return Array(0,0)
        if (nums.length == 1 && nums(0) != target) return Array(-1,-1)
        var left = 0
        var right = nums.length - 1
        while (left < right){
            var  mid = (left+right)/2
            if (nums(mid) < target) left = mid + 1
            else right =  mid
        }
        println(left)
        println(right)
        if (left == right && nums(left) == target) {
            while((left-1 >= 0) && nums(left-1) == target) left-=1
            while((right+1 <= nums.length-1) && (nums(right+1) == target)) right+=1
            return Array(left, right)
        }
        else return Array(-1, -1) 
    }
}
posted @ 2020-05-08 21:53  ganshuoos  阅读(113)  评论(0编辑  收藏  举报