问题:统计一个数字在排序数组中出现的次数。例如输入排序数组{1,2,3,3,3,3,4,5}和数字3,
由于3在这个数组中出现了4次,因此输出4。
思路1:因为数组是排序的,所以我们可以利用折半查找来找到我们要找的这个数字。
找到这个数字之后,从左右两边依次开始继续检测,找到最左边和最右边的那个数字,相减就可以得到次数。
但是这样做的时间复杂度为O(n),
思路2:利用折半查找找到我们要找的元素之后,继续使用折半查找找到第一个和最后一个我们需要找到的元素。
int getFirstK(int *a, int length, int start, int end, int k)
{
if(start > end)
return -1;
int mid = (end + start) / 2;
int midData = a[mid];
if(midData == k)
{
if(mid == 0 || a[mid-1] != k)
return mid;
else
end = mid - 1;
}
else if(midData > k)
end = mid - 1;
else
start = mid + 1;
return getFirstK(a, length, start, end, k);
}
int getLastK(int *a, int length, int start, int end, int k)
{
if(start > end)
return -1;
int mid = (end + start) / 2;
int midData = a[mid];
if(midData == k)
{
if(mid == length - 1 || a[mid+1] != k)
return mid;
else
start = mid + 1;
}
else if(midData > k)
end = mid - 1;
else
start = mid + 1;
return getLastK(a, length, start, end, k);
}
int getResult(int *a, int length, int start, int end, int k)
{
int first = getFirstK(a, length, start, end, k);
int last = getLastK(a, length, start, end, k);
if(first < 0 || last < 0)
return -1;
return last - first + 1;
}