题目描述
统计一个数字k在排序数组中出现的次数。
基本思路
由于是已排序的数组,所以可以通过二分法寻找到 k。而所有的 k 都是聚集在一起的,那么找到一个 k 就可以分别向左和向右搜索,就可以获取到 k 的范围了。
以下为具体解答:
public int GetNumberOfK(int [] array , int k) {
int index = binarySearch(array, k);
if (index == -1) {
return 0;
}
int left = index;
while (left - 1 >= 0 && array[left-1] == k) {
left = left - 1;
}
int right = index;
while (right + 1 < array.length && array[right+1] == k) {
right = right + 1;
}
return right - left + 1;
}
private int binarySearch(int[] array, int k) {
if (array == null) {
return -1;
}
int start = 0;
int end = array.length - 1;
while (start <= end) {
int mid = (start + end) / 2;
if (array[mid] == k) {
return mid;
} else if (array[mid] > k) {
end = mid - 1;
} else {
start = mid + 1;
}
}
return -1;
}