题目描述
统计一个数字在排序数组中出现的次数。
- 解题思路:
遇到排序数组,优先无脑考虑二分法。找出这个数字的左边位置,再找出这个数字的右边位置,两个位置的距离就是出现的次数。复杂度是O(logn)。 - 注意点
注意边界判断,即没有找到的时候返回的是-1,此时应特殊输出0。
【考点】 查找 二分法
class Solution {
public:
int GetNumberOfK(vector<int> data, int k) {
int length = data.size();
int firstIndex = getFirstK(data, k, 0, length - 1);
int lastIndex = getLastK(data, k, 0, length - 1);
if (firstIndex == -1 || lastIndex == -1) return 0;
int res = lastIndex - firstIndex + 1;
return res;
}
int getFirstK(vector<int> data, int k, int low, int high) {
if (low > high)
return -1;
int mid = (low + high) / 2;
if (data[mid] == k)
{
if ((mid > 0 && data[mid - 1] != k) || mid == 0)
return mid;
else
high = mid - 1;
}
else if (data[mid] > k)
high = mid - 1;
else
low = mid + 1;
return getFirstK(data, k, low, high);
}
int getLastK(vector<int> data, int k, int low, int high) {
if (low > high)
return -1;
int mid = (low + high) / 2;
if (data[mid] == k)
{
if ((mid < data.size()-1 && data[mid + 1] != k) || mid == data.size()-1)
return mid;
else
low = mid + 1;
}
else if (data[mid] > k)
high = mid - 1;
else
low = mid + 1;
return getLastK(data, k, low, high);
}
};