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)