【问题】统计一个数字在排序数组中出现的次数。
【思路】首先要清楚这种看似简单的题目,使用直接遍历是可以,但一般不得分,由于题目给出了排序数组,对于排序数组来说,常用的搜索查找方式为二分查找(binary search)。这里有个巧妙的方法,我们并不是去搜索k这个数,而是去搜索k-0.5和k+0.5这两个小数,进而返回待插入的位置!
比如:1 2 2 2 3 4 且k = 2
则,k+0.5会返回3的索引即4,而k-0.5会返回第一个2的索引即1,两者相减得3,即为最后的结果!二分代码如下,只返回begin位置!
class Solution { public: int GetNumberOfK(vector<int> data ,int k) { return biSearch(data, k+0.5) - biSearch(data, k-0.5); } private: int biSearch(vector<int>& data, double num){ int begin = 0, end = data.size() - 1; while(begin <= end){ int mid = begin + (end - begin) / 2; if(data[mid] < num) begin = mid + 1; else if(data[mid] > num) end = mid - 1; } return begin; } };
另外一个思路直接使用STL中的库函数equal_range来获取与某一个值相等的上下边界,十分好用的!
class Solution { public: int GetNumberOfK(vector<int> data ,int k) { auto res = equal_range(data.begin(), data.end(), k); return res.second - res.first; } };