题目:统计一个数字在排序数组中出现的次数。例如,输入一个排序的数组1,2,3,4,4,4,4,4,5,6
和数字4,则输出4出现的次数5。
解题思路一: 我们可以根据二分查找找到要查找数字在数组中第一次和最后一次出现的位置,就可以确定总共出现过几次了。但是要怎么找呢?首先根据二分查找找到一个要查找的数字,然后在它的左右两边顺序查找到它们第一次和最后一次出现的位置
。但是当要查找的数字在左边或者右边全都是,那么这种方法的时间复杂度是O(n)
,和顺序查找的时间复杂度一样
,肯定不会满足面试官的要求。
解题思路二:依旧是二分查找的思想。
- 1.根据
二分查找
找到第一个
所给数字 - 2.判断如果这个数
是数组的第一个数字
或者这个数字前面那个数字和这个数字不相同
,则这个数字就是第一个
。否则第一次出现的数在中间的左边
,去左边查找。 - 3.判断如果这个数
是数组的最后一个数字
或者这个数字后面那个数字和这个数字不相同
,则这个数字就是最后一次出现的
。否则后一次出现的的数在中间的右边,去右边查找。
代码实现:
//查找数组中第一个K的下标(二分查找的思想,循环实现)
int GetFirstPos(int *arr, int k, int len, int start, int end)
{
assert(arr);
while (start <= end)
{
int mid = start + (end - start) / 2;
if (arr[mid] == k)
{
//如果中间数字为K,查看前一个数字是否为K或者是否为第一个数字
if ((mid == 0) || (mid > 0 && arr[mid - 1] != k))
return mid;
else
end = mid - 1;
}
else if (arr[mid] > k)
end = mid - 1;
else
start = mid + 1;
}
return -1;
}
//查找数组中第一个K的下标(二分查找的思想,递归实现)
int GetLastPos(int *arr, int k, int len, int start, int end)
{
assert(arr);
if (end < start)
return -1;
int mid = start + (end - start) / 2;
if (arr[mid] == k)
{
//如果中间数字为K,查看后一个数字是否为K或者是否为最后一个数字
if ((mid == len - 1) || (mid <len - 1 && arr[mid + 1] != k))
return mid;
else
start = mid + 1;
}
else if (arr[mid] > k)
end = mid - 1;
else
start = mid + 1;
return GetLastPos(arr, k, len, start, end);
}
//递增有序的数组中K的个数
int CountOfNumber(int *arr, int k, int len)
{
assert(arr);
if (len <= 0)
return -1;
int start = 0;
int end = len - 1;
int first = GetFirstPos(arr, k, len, start, end);
int last = GetLastPos(arr, k, len, start, end);
if (first > -1 && last > -1)
return last - first + 1;
}