JavaScript——leetcode剑指offer 53 - I. 在排序数组中查找数字 I

JavaScript——leetcode剑指offer 53 - I. 在排序数组中查找数字 I

题目描述

统计一个数字在排序数组中出现的次数。

示例 1:
输入: nums = [5,7,7,8,8,10], target = 8
输出: 2

示例 2:
输入: nums = [5,7,7,8,8,10], target = 6
输出: 0

代码片

/**
 * @param {number[]} nums
 * @param {number} target
 * @return {number}
 */

//定义一个binarySearch表示在nums数组中二分查找target的位置，
const binarySearch = (nums, target, lower) => {
    let left = 0; right = nums.length - 1, ans = nums.length;
    while(left <= right) {
        const mid = Math.floor((left + right) / 2);
        if(nums[mid] > target || (lower && nums[mid] >= target)) {
            right = mid - 1;
            ans = mid;
        } else {
            left = mid + 1;
        }
    }
    return ans;
}

//考虑target在数组中出现的次数
//第一个等于target的位置记为leftIdx
//第一个大于target的位置减一，记为rightIdx
//当target在数组中存在时，target在数组中出现的次数为rightIdx-leftIdx+1
var search = function(nums, target) {
    let ans = 0;
    //如果lower为true，则查找第一个大于等于target的下标,否则查找第一个大于target的下标
    const leftIdx = binarySearch(nums, target, true);
    const rightIdx = binarySearch(nums, target, false) - 1;
    if(leftIdx <= rightIdx && rightIdx < nums.length && nums[leftIdx] === target && nums[rightIdx] === target) {
        ans = rightIdx - leftIdx + 1;
    }
    return ans;
};