题目描述
统计一个数字在排序数组中出现的次数。
题目分析
这题用暴力解也可以过,不过面试官肯定期待更好的解法。
查找我们最熟悉的就是二分查找了,不过二分查找查找的数在数组中只有一个,我们这里却有很多个,是一段。所以我们要确定找的数的开始位置和结束位置
主要就是判断的条件,当a[mid]的值等于k时,指针该如何走。
代码
function GetNumberOfK(data, k) { if (getEnd(data, k) === -1 && getBegin(data, k) === -1) return 0; return getEnd(data, k) - getBegin(data, k) + 1; } function getBegin(data, k) { let [left, right] = [0, data.length - 1]; let mid = left + right >> 1; while (left <= right) { if (data[mid] > k) { right = mid - 1; } else if (data[mid] < k) { left = mid + 1; } else if (mid - 1 >= 0 && data[mid - 1] === k) { right = mid - 1; } else return mid; mid = left + right >> 1; } return -1; } function getEnd(data, k) { let [left, right] = [0, data.length - 1]; let mid = left + right >> 1; while (left <= right) { if (data[mid] > k) { right = mid - 1; } else if (data[mid] < k) { left = mid + 1; } else if (mid + 1 < data.length && data[mid + 1] === k) { left = mid + 1; } else return mid; mid = left + right >> 1; } return -1; }