JS算法題之每日一題- 17.在排序數組中查找元素的第一個和最後一個位置

微信公衆號： 醉前端,關注可瞭解更多題目。

題 目

Q:給定一個按照升序排列的整數數組 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]

答 案

const searchRange = (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];
};

const 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;
    }
}