数字在排序数组中出现的次数

题目描述

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

public class Solution {
    public int GetNumberOfK(int [] array , int k) {

       int index = bs(array, k, 0, array.length - 1);
       int cnt = 0;


       for(int i=index; i>=0 && i<array.length; i--)
           if(array[i] == k)
               cnt++;
        for(int i=index+1; i<array.length; i++)
           if(array[i] == k)
               cnt++;  

        return cnt;
    }

    public int bs(int [] array , int k, int lo, int hi) {

        while(lo <= hi) {
            int mid = lo + (hi - lo) / 2;

            int cmp = array[mid] - k;
            if(cmp > 0) 
                hi = mid - 1;
            else if(cmp < 0)
                lo = mid + 1;
            else
                return mid;
        }
        return lo;
    }
}

尽量不去修改bs中的内容，当找不到是在for语句 中添加约束条件i>=0 && i<array.length 来处理没有找到的情况，并防止抛出异常