题目描述
统计一个数字在排序数组中出现的次数。
题目思路:
1)常规思路:对该数字在数组中遍历,计数出现次数
2)改进:既然是一个排序数组,很自然就可以想到二分查找。
用二分查找分别定位数字k在数组中第一次出现和最后一次出现的位置。
(1)为了获得数字k最后一次出现的位置,我们可以将数组中所有小于等于k的都归为左边部分,left = mid + 1,指向第一个大于k的位置。
(2)为了获得数字k第一次出现的位置,我们可以将数组中所有大于等于k的都归为右边部分,right = mid,指向k第一次出现的位置。
另外需要注意的是,代码中需要判断数字k是否在数组中存在。
class Solution {
public:
int GetNumberOfK(vector<int> data, int k) {
int len = data.size();
if (len == 0) return 0;
int right = GetRightIndex(data, k);
int left = GetLeftIndex(data, k);
if (right == -1 || left == -1) return 0;
return right - left + 1;
}
int GetRightIndex(vector<int> data, int k) {
int left = 0; int right = data.size();
while (left<right) { //目的:将数组中所有小于等于k的数字都放在左边部分中。
// 最终的定位是:left指向第一个大于k的位置。
int mid = left + ((right - left) >> 1);
if (data[mid] > k) {
right = mid;
}
else {
left = mid + 1;
}
}
if(left-1>=0 && data[left-1] == k) //需要考虑数字不存在的情况。
return left - 1;
return -1;
}
int GetLeftIndex(vector<int> data, int k) {
int left = 0; int right = data.size();
while (left < right) { //目的:将所有大于等于k的数组都放在右边部分。
//最终的定位是: right指向k第一次出现的位置
int mid = left + ((right - left) >> 1);
if (data[mid] >= k) {
right = mid;
}
else {
left = mid + 1;
}
}
if(right>=0 && right < data.size() && data[right] == k)
return right;
return -1;
}
};