题目
统计一个数字在排序数组中出现的次数。
例如
输入排序数组{1,2,3,3,3,3,4,5}和数字3
输出4
思路
思路一
遍历整个数组,统计其出现的次数,时间复杂度O(n)
思路二
根据排序数组这个题设,可以想到用二分查找的方法。
如果中间位置的值小于目标值,在数组右部分继续查找。
如果中间位置的值大于目标值,在数组左边部分继续查找。
如果中间位置的值等于目标值且左边的值也等于目标值,在数组左边部分继续查找。
如果中间位置的值等于目标值且左边的值不等于目标值,返回左边第一个位置。
同理,右边最后一个位置。
根据左右出现的位值来计算该数出现的次数。
时间复杂度为O(lgn)
代码
public static int getKNum(int[] numbers,int k){
int number = 0;
if(numbers == null)return 0;
int left = getFirstK(numbers,k,0,numbers.length - 1);
int right = getLastK(numbers,k,0,numbers.length - 1);
if(left > -1 && right > -1){
number = right - left + 1;
}
return number;
}
public static int getFirstK(int[] numbers,int k,int start,int end){
if(start > end)return - 1;
int midIndex = (start + end) / 2;
int midData = numbers[midIndex];
if(midData == k){
if(midIndex > 0 && numbers[midIndex - 1] != k || midIndex == 0){
return midIndex;
}else {
end = midIndex - 1;
}
} else if(midData > k){
end = midIndex - 1;
}else {
start = midIndex + 1;
}
return getFirstK(numbers,k,start,end);
}
public static int getLastK(int[] numbers,int k,int start,int end){
if(start > end)return -1;
int midIndex = (start + end) / 2;
int midData = numbers[midIndex];
if(midData == k){
if(midIndex < numbers.length - 1 && numbers[midIndex + 1] != k || midIndex == numbers.length - 1){
return midIndex;
}else{
start = midIndex + 1;
}
}else if(midData > k){
end = midIndex - 1;
}else {
start = midIndex + 1;
}
return getLastK(numbers,k,start,end);
}
总结
看到排序的数组就该想到二分查找