leetcode刷题笔记三十四 在排序数组中查找元素的第一个和最后一个位置
问题描述:
给定一个按照升序排列的整数数组 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)
}
}