34.在排序數組中查找元素的第一個和最後一個位置--二分查找

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]

來源:https://leetcode-cn.com/problems/find-first-and-last-position-of-element-in-sorted-array/

題解
二分查找,結果就是第一個大於等於target的位置第一個大於target的位置-1

var searchRange = function (nums, target) {
    let x = 0, y = nums.length - 1;
    /* 找出第一個大於等於target的位置 */
    while (x <= y) {/* 退出條件x=y+1 */
        let m = x + Math.floor(y - x >> 1);
        /* 只要中間值大於target,不斷往左邊走,
           y最終就是最後一個小於target的位置 */
        if (nums[m] >= target) y = m - 1;
        /* 遇到小於往右邊走,x最終就是第一個大於等於target的位置 */
        else x = m + 1;
    }
    let res1 = nums[x] === target ? x : -1;

    /* 找出第一個大於target的位置 */
    x = 0, y = nums.length - 1;
    while (x <= y) {/* 退出條件x=y+1 */
        let m = x + Math.floor(y - x >> 1);
        /* 只要中間值大於target,不斷往左邊走,
           y最終就是最後一個小於等於target的位置 */
        if (nums[m] > target) y = m - 1;
        /* 遇到小於往右邊走,x最終就是第一個大於target的位置 */
        else x = m + 1;
    }
    let res2 = nums[y] == target ? y : -1;
    return [res1, res2];
};
let ans = searchRange([5, 7, 7, 8, 8, 10], 8)
console.log(ans)
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章