题目:统计一个数字在排序数组中出现的次数。例如输入排序数组{ 1,2,3,3,3,3,4,5}和数字3,。由于3在这个数组中出现了4次,因此输出4
方法一:时间复杂度为O(N)
思路:直接把数组遍历一遍即可,简单粗暴
代码:
int GetNumCount(int* arr, int sz, int key)
{
assert(arr);
int count = 0;
for (int i = 0; i < sz; i++)
{
if (arr[i] == key)
{
count++;
}
}
return count;
}
方法二:时间复杂度为O(LogN)
思路:对于一个有序数组的话,我们一般反应就是采用二分查找就行了。
对于这道题可以采用二分查找,找到我们我们要查找的数字,但是如何判断其在数组中出现几次呢?
我们可以这样做:
思路:采用二分查找找到第一次该数字在数组中出现的位置和最后一次出现的位置,然后再将其相减就可以了
1.求第一次在数组中出现的位置,找到的话返回下标值,否则返回-1
2.求第最后一次在数组中出现的位置,找到的话返回下标值,否则返回-1
3.对其进行判断如果都是-1的话,说明没有出现,返回0,否则返下标值相减再加1
代码:
//二分查找第一个出现的数字的下标,时间复杂度为O(logN)
int GetFristNum(int* arr, int onset, int finish, int key)
{
assert(arr);
int start = onset;
int end = finish;
while (start <= end)
{
int mid = start + ((end - start) >> 1);
if (arr[mid] == key)
{
if (arr[mid - 1] != key)
{
return mid;
}
else
{
return GetFristNum(arr, start, mid - 1, key);
}
}
else if (arr[mid] < key)
{
start = mid + 1;
}
else
{
end = mid - 1;
}
}
return -1;
}
//二分查找最后一个出现的数字的下标,时间复杂度为O(logN)
int GetEndNum(int* arr, int onset, int finish, int key)
{
assert(arr);
int start = onset;
int end = finish;
while (start <= end)
{
int mid = start + ((end - start) >> 1);
if (arr[mid] == key)
{
if (arr[mid + 1] != key)
{
return mid;
}
else
{
return GetEndNum(arr, mid + 1, finish, key);
}
}
else if (arr[mid] < key)
{
start = mid + 1;
}
else
{
end = mid - 1;
}
}
return -1;
}
//总的时间复杂度为O(logN)
int GetNumCount(int* arr, int sz, int key)
{
assert(arr);
//如果在数组中没有找到出现的数字,那么就返回0
if (GetEndNum(arr, 0, sz - 1, key) == -1 || GetFristNum(arr, 0, sz - 1, key) == -1)
{
return 0;
}
return (GetEndNum(arr, 0, sz - 1, key) - GetFristNum(arr, 0, sz - 1, key) + 1);
}
由于查找最后一次出现的下标和第一次出现的下标所用的时间复杂度都是O(LogN),所以总体的时间复杂度为O(LogN).
相比第一种方法时间上效率更高。