【剑指offer】数字在排序数组中出现的次数(Java解答)

题目描述
统计一个数字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;
}

你可能感兴趣的:(剑指Offer)