剑指offer系列——37.数字在排序数组中出现的次数

Posted 2021-03-09 xym4869

Q：统计一个数字在排序数组中出现的次数。
A：
1.用STL库函数解决
（1）count函数：algorithm头文件定义，其功能类似于find。这个函数使用一对迭代器和一个值做参数，返回这个值出现次数的统计结果。

public:
    int GetNumberOfK(vector<int> data ,int k) {
        return count(data.begin(),data.end(),k);
    }

或
（2）equal_range函数：函数equal_range()返回first和last之间等于val的元素区间.返回值是一对迭代器。此函数假定first和last区间内的元素可以使用<操作符或者指定的comp执行比较操作.

class Solution {
public:
    int GetNumberOfK(vector<int> data ,int k) {
        auto resultPair = equal_range(data.begin(), data.end(),k);
        return resultPair.second - resultPair.first;
    }
};

2.二分查找（源于@披萨大叔）
P.S.我有一个疑问，题目提的是排序，没说升序还是降序……二分法不需要确定一下吗？？

public class Solution {
    public int GetNumberOfK(int [] array , int k) {
        int length = array.length;
        if(length == 0){
            return 0;
        }
        int firstK = getFirstK(array, k, 0, length-1);
        int lastK = getLastK(array, k, 0, length-1);
        if(firstK != -1 && lastK != -1){
             return lastK - firstK + 1;
        }
        return 0;
    }
    //递归写法
    private int getFirstK(int [] array , int k, int start, int end){
        if(start > end){
            return -1;
        }
        int mid = (start + end) >> 1;
        if(array[mid] > k){
            return getFirstK(array, k, start, mid-1);
        }else if (array[mid] < k){
            return getFirstK(array, k, mid+1, end);
        }else if(mid-1 >=0 && array[mid-1] == k){
            return getFirstK(array, k, start, mid-1);
        }else{
            return mid;
        }
    }
    //循环写法
    private int getLastK(int [] array , int k, int start, int end){
        int length = array.length;
        int mid = (start + end) >> 1;
        while(start <= end){
            if(array[mid] > k){
                end = mid-1;
            }else if(array[mid] < k){
                start = mid+1;
            }else if(mid+1 < length && array[mid+1] == k){
                start = mid+1;
            }else{
                return mid;
            }
            mid = (start + end) >> 1;
        }
        return -1;
    }