数字在排序数组中出现的次数

最新推荐文章于 2022-11-13 17:16:41 发布

转载最新推荐文章于 2022-11-13 17:16:41 发布 · 122 阅读

本文介绍了一种高效算法，用于统计一个数字在已排序数组中出现的次数。通过二分查找法找到目标数字的第一个和最后一个位置，进而计算其出现次数。适用于大数据集和实时查询场景。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目：统计一个数字在排序数组中出现的次数。例如输入排序数组{1, 2, 3, 3,3, 3, 4, 5}和数字3，由于3在这个数组中出现了4次，因此输出4。

int GetNumberOfK(int* data, int length, int k)
{
	int number = 0;
	if (data != nullptr || 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)
			number = last - first + 1;
	}
	return number;
}
//找到第一个K
int getFirstK(int* data, int length, int k, int start, int end)
{
	if (start > end)
		return -1;
	int middleIndex = (start + end) / 2;
	int middleData = data[middleIndex];

	if (middleData < k)
		start = middleIndex + 1;
	else if (middleData > k)
		end = middleIndex - 1;
	else if (middleData == k)
	{
		if ((middleIndex > 0 && data[middleIndex - 1] != k) || middleIndex == 0)
			return middleIndex;
		else
			end = middleIndex - 1;
	}
	return getFirstK(data, length, k, start, end);
}
//找到最后一个K
int getLastK(int* data, int length, int k, int start, int end)
{
	if (start > end)
		return -1;
	
	int middleIndex = (start + end) / 2;
	int middleData = data[middleIndex];

	if (middleData < k)
		start = middleIndex + 1;
	else if (middleData > k)
		end = middleIndex - 1;
	else if (middleData == k)
	{
		if ((middleData < length - 1 && data[middleIndex + 1] != k) || middleIndex == length - 1)
			return middleIndex;
		else
			start = middleIndex + 1;
	}
	return getLastK(data, length, k, start, end);
}