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

2024年6月9日 101次阅读

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

思路：
最直观的方法就是从头遍历，统计次数。

class Solution {
public:
    int GetNumberOfK(vector<int> data ,int k) {
        if(data.size()==0)
            return 0;

        int count = 0;
        for(int i=0; i<data.size(); ++i){
            if(k == data[i])
                count++;
        }
        return count;
    }
};

也可以借助库函数

  • lower_bound()函数是返回第一个大于等于x元素的地址
  • upper_bound()函数是返回第一个大于x元素的地址

举个例子，1 2 2 2 5 6 7
第一个大于等于2的地址是第一个2的地址
第一个大于2的地址是5的地址
相减刚好是2的个数

class Solution {
public:
    int GetNumberOfK(vector<int> data ,int k) {
        if(data.size()==0)
            return 0;

        auto length1 = std::lower_bound(data.begin(), data.end(), k);//返回第一个大于等于k的位置
        auto length2 = std::upper_bound(data.begin(), data.end(), k);//返回第一个大于k的位置
        return (length2-length1);
    }
};

或者用二分法进行查找，找到k的所对应的下标后，定义两个下标变量，一个向前走，一个向后走，最后取两个下标的差值减1就得到了该数出现的次数。

class Solution {
public:
    int GetNumberOfK(vector<int> data ,int k) {
        if(data.size()==0)
            return 0;

        int left = 0;
        int right = data.size()-1;
        int mid;
        int i=-1;
        //利用二分法查找到数字的位置
        while(left <= right){
            mid = (left+right)/2;
            if(data[mid] == k){
                i = mid;
                break;
            }
            else if(k < data[mid])
                right = mid-1;
            else
                left = mid+1; 
        }
        if(i == -1)
            return 0;
        left = i-1;//记录该数字出现的首位置
        right = i+1;//记录该数字出现的末位置
        while(left>=0 && data[left]==k){
          --left;
        }
        while(right<data.size() && data[right]==k){
          ++right;
        }

        return right-left-1;
    }
};
点赞