题目描述
统计一个数字在排序数组中出现的次数。
解法1:
普通解法-顺序查找:时间复杂度:O(n)
解法2:
排序数组的查找可以用二分法,怎么用?有多个重复的数字,若是查找的中间数和两边的都相同,那么两边都要找时间复杂度:O(n)。没有利用二分查找的优势
思路转换:将查找的数字转换为查找第一个数字和最后一个数字,依然用二分法查找
时间复杂度:O(logn)
二分法的改进
查找第一个数字k
1)中间的数比k大,下一次在前半段查找
2)中间的数比k小,下一次在后半段查找
3)中间的数字等于k,那么k再和前一个数字比较,若等于k,那么往前半段查找,若不等于k,那么返回
查找最后一个数字k
1)中间的数比k大,下一次在前半段查找
2)中间的数比k小,下一次在后半段查找
3)中间的数字等于k,那么k再和后一个数字比较,若等于k,那么往后半段查找,若不等于k,那么返回
class Solution
{
public:
int GetNumberOfK(vector<int> array , int k)
{
int length = array.size();
if(length == 0){
return 0;
}
int firstK = getFirstK(array, k, 0, length-1);
int lastK = getLastK(array, k, 0, length-1);
if(firstK != -1 && lastK != -1){
return lastK - firstK + 1;
}
return 0;
}
//递归写法
private:
int getFirstK(vector<int> array , int k, int start, int end)
{
if(start > end){
return -1;
}
int mid = (start + end) >> 1;
if(array[mid] > k){
return getFirstK(array, k, start, mid-1);
}else if (array[mid] < k){
return getFirstK(array, k, mid+1, end);
}else if(mid-1 >=0 && array[mid-1] == k){
return getFirstK(array, k, start, mid-1);
}else{
return mid;
}
}
//循环写法
private:
int getLastK(vector<int> array , int k, int start, int end)
{
int length = array.size();
int mid = (start + end) >> 1;
while(start <= end){
if(array[mid] > k){
end = mid-1;
}else if(array[mid] < k){
start = mid+1;
}else if(mid+1 < length && array[mid+1] == k){
start = mid+1;
}else{
return mid;
}
mid = (start + end) >> 1;
}
return -1;
}
};
参考:【1】剑指
【2】https://www.nowcoder.com/profile/7221411/codeBookDetail?submissionId=14533282