【剑指offer】在排序数组中查找数字（二分查找）

题目描述

统计一个数字在排序数组中出现的次数。例如，对排序数组{1，2，3，3，3，3，4，5}和数字3，由于3在这个数组中出现了4次，因此输出4。

思路

对于排序数组中查找，首先考虑使用二分查找。该题目要求统计元素出现的次数，则可以分解以下步骤：
（1）找到该元素在排序数组中最左边的位置leftIndex
（2）找到该元素在排序数组中最右边的位置rightIndex
（3）计算得到出现次数为rightIndex-leftIndex+1
注意：该题的关键在于找到最左（右）边位置的处理
时间复杂度：O（logn）

代码

/*
 * 时间复杂度O(logn) 空间复杂度O(1)
 */
class Solution {
public:
    int getNumberOfK(vector<int> nums, int k){
        if(nums.empty()) return 0;
        int ret = 0;
        int left = 0, right = nums.size()-1;

        int leftIndex = -1;  // k元素最左位置
        int rightIndex = -1; // k元素最右位置
        // 查找最左边的k元素位置
        while(left <= right){
            int mid = (left+right)/2;
            if(nums[mid]<k)
                left = mid+1;
            else if(nums[mid] >k)
                right = mid -1;
            else{
                if((mid > 0 && nums[mid-1] != k )|| mid == 0){
                    leftIndex = mid;
                    break;
                }else
                    right = mid -1;
            }
        }

        if(leftIndex == -1) return 0;    // 没有出现

        // 查找最右边的k元素位置
        left = 0, right = nums.size()-1;
        while(left <= right){
            int mid = (left+right)/2;
            if(nums[mid]<k)
                left = mid+1;
            else if(nums[mid] >k)
                right = mid -1;
            else{
                if((mid < nums.size()-1 && nums[mid+1] != k) || mid == nums.size()-1){
                    rightIndex = mid;
                    break;
                }else
                    left = mid + 1;
            }
        }
        return rightIndex - leftIndex + 1;
    }
};