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

题目描述

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

思路：记住二分搜索的复杂度是O(logn)，复杂度的计算方式是主定理。二分搜索模板要记得。

这题要注意没有找到相关元素的返回情况，初始化的时候将两个位置初始化为-1，在最后判断以下，如果两个都是-1就是没有找到位置，这个时候就返回0，如果找到了就pos2 - pos1 + 1.

 

class Solution {
public:
    int GetNumberOfK(vector<int> data ,int k) {
        if(data.size() == 0){
            return 0;
        }
        int start = 0,end = data.size() - 1;
        int mid = start + (end - start) / 2;
        int pos1 = -1,pos2 = -1;
        int flag = false;
        while(start + 1 < end){
            mid = start + (end - start) / 2;
            if(data[mid] == k){
                end = mid;
            }
            else if(data[mid] < k){
                start = mid;
            }
            else{
                end = mid;
            }            
        }
        if(data[start] == k){
            pos1 = start;
        }
        else if(data[end] == k){
            pos1 = end;
        }
        start = 0,end = data.size() - 1;
        while(start + 1 < end){
            mid = start + (end - start) / 2;
            if(data[mid] == k){
                start = mid;
            }
            else if(data[mid] < k){
                start = mid;
            }
            else{
                end = mid;
            }            
        }
        if(data[end] == k){
            pos2 = end;
        }
        else if(data[start] == k){
            pos2 = start;
        }
        int cnt = 0;
        if(pos2 != -1 && pos1 != -1){//判断是否找到该元素
            cnt = pos2 - pos1 + 1;
        }            
        return cnt;
    }
};