题目:统计一个数字在排序数组中出现的次数。例如输入排序数组{1,2,3,3,3,3,4,5}和数字3,由于3在这个数组中出现了4次,因此输出4。
看到题目的第一感觉我们肯定想到的是遍历整个数组来求解,但这样的算法时间复杂度是O(N),还不是最优解,我们可以想到使用二分查找来优化。
由于数组是排好序的,假设我们是统计数字K在排序数组中出现的次数,所以我们可以用二分查找直接找到数组中的第一个K和最后一个K。那么如何用二分查找算法在数组中找到第一个K呢?二分查找算法总是先拿中间的数字和K作比较。如果中间的数字比K大,那么K只有可能出现在数组的前半段,下一轮我们只在数组的前半段查找就可以了,如果中间的数字比K小,那么K只有可能出现在数组的后半段,下一轮我们只在数组的后半段查找就可以了。如果中间的数字和K相等呢?我们先判断这个数字是不是第一个K。如果位于中间数字的前面一个数字不是K,此时中间的数字刚好就是第一个K.如果中间数字的前面一个数字也是K,也就是说第一个K肯定在数组的前半段,下一轮我们仍然需要在数组的前半段查找。找到数组中的最后一个K的思路与上述类似。
int GetFirstK(int* data, int length, int k, int start, int end)
{
if (start > end)
return -1;
int middleIndex = start +( (end - start) >> 1);
int middleData = data[middleIndex];
if (middleData == k)
{
if ((middleIndex > 0 && data[middleIndex - 1] != k)|| middleIndex == 0)
{
return middleIndex;
}
else
{
end = middleIndex - 1;
}
}
else if (middleData > k)
{
end = middleIndex - 1;
}
else
{
start = middleIndex + 1;
}
return GetFirstK(data, length, k, start, end);
}
int GetLastK(int* data, int length, int k, int start, int end)
{
if (start > end)
return -1;
int middleIndex = start +( (end - start) >> 1);
int middleData = data[middleIndex];
if (middleData == k)
{
if ((middleIndex < length - 1 && data[middleIndex + 1] != k) || middleIndex == length - 1)
return middleIndex;
else
start = middleIndex + 1;
}
else if (middleData < k)
{
start = middleIndex + 1;
}
else
{
end = middleIndex - 1;
}
return GetLastK(data, length, k, start, end);
}
int GetNumOfK(int* data, int length, int k)
{
int num = 0;
if (data != NULL && length > 0)
{
int first = GetFirstK(data, length, k, 0, length - 1);
int last = GetLastK(data, length, k, 0, length - 1);
if (first > -1 && last > -1)
num = last - first + 1;
}
return num;
}
int main()
{
int data[] = { 1, 2, 3, 3, 3, 3, 4, 5 };
int len = sizeof(data) / sizeof(data[0]);
int ret = GetNumOfK(data, len, 3);
cout << ret << endl;
system("pause");
return 0;
}