LeetCode34.在排序數組中查找元素的第一個和最後一個位置 JavaScript

給定一個按照升序排列的整數數組 nums,和一個目標值 target。找出給定目標值在數組中的開始位置和結束位置。javascript

你的算法時間複雜度必須是 O(log n) 級別。java

若是數組中不存在目標值,返回 [-1, -1]算法

示例 1:數組

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

示例 2:spa

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

答案參考:code

/**
 * @param {number[]} nums
 * @param {number} target
 * @return {number[]}
 */
var searchRange = function (nums, target) {
    let targetIndex = binarySearch(nums, target, 0, nums.length - 1)
    if (targetIndex == -1) return [-1, -1]
    let l = targetIndex, r = targetIndex
    while(l > 0 && nums[l - 1] == target){
        l--
    }
    while(r < nums.length - 1 && nums[r + 1] == target){
        r++
    }
    return [l, r]
};

function binarySearch(arr, val, lo, hi) {
    if (hi < lo) return -1
    let mid = lo + parseInt((hi - lo) / 2)

    if (val < arr[mid]) {
        return binarySearch(arr, val, lo, mid - 1)
    } else if (val > arr[mid]) {
        return binarySearch(arr, val, mid + 1, hi)
    } else {
        return mid
    }
}

相關文章
相關標籤/搜索