题目
统计一个数字在排序数组中出现的次数。
例如输入排序数组[1, 2, 3, 3, 3, 3, 4, 5]和数字3,由于3在这个数组中出现了4次,因此输出4。
样例
输入:[1, 2, 3, 3, 3, 3, 4, 5] , 3
输出:4

解法:二分

数组已经排好序,那么直接分别查找上下界即可
时间复杂度O(logn),空间复杂度O(1)

  1. class Solution {
  2. public:
  3. int getNumberOfK(vector<int>& nums , int k) {
  4. if (nums.empty()) return 0;
  5. if (k < nums[0] || k > nums[nums.size() - 1]) return 0;
  6. int l = 0, r = nums.size() - 1;
  7. while (l < r) {
  8. int mid = l + r + 1 >> 1;
  9. if (nums[mid] <= k) l = mid;
  10. else r = mid - 1;
  11. }
  12. int right = l;
  13. l = 0, r = nums.size() - 1;
  14. while (l < r) {
  15. int mid = l + r >> 1;
  16. if (nums[mid] >= k) r = mid;
  17. else l = mid + 1;
  18. }
  19. int left = l;
  20. return right - left + 1;
  21. }
  22. };