题目描述
统计一个数字在排序数组中出现的次数

看到这题的第一反应就是暴力求解,当然了,这不失为一种可行的方法,但是大多数情况下,暴力求解会因为速度限制而过不了testcases,所以我们需要进行优化。
而这题里有一个重点,那就是 排序 ,我们就可以立马联想到二分法求解,这样我们的速度就会变成O(nlogn)。
所以这题分三步:

  1. 二分法找出任何一个等于k的数组元素,记录下他的index
  2. 根据这个index向前查找所有等于k的元素数量
  3. 根据这个index向后查找所有等于k的元素数量

具体代码如下:

public class Solution {
    public int GetNumberOfK(int [] array , int k) {
        if(array.length == 0 || k < array[0] || k > array[array.length-1]){
            return 0;
        }
        int left = 0;
        int right = array.length -1;
        int count = 0;
        int found = 0;
        int mid = -1;
        while(left < right){
            mid = (left+right)/2;
            if(array[mid] > k){
                right = mid-1;
            }else if(array[mid] < k){
                left = mid+1;
            }else{
                count++;
                found = mid;
                break;
            }
        }

        int prev = mid-1;
        int foll = mid+1;
        while(prev >= left){
            if(array[prev] == k){
                count++;
                prev--;
            }else{
                break;
            }
        }

        while(foll <= right){
            if(array[foll] == k){
                count++;
                foll++;
            }else{
                break;
            }
        }
        return count;
    }
}