问:统计一个数字在排序数组中出现的次数。
//此题需要找到在排序数组中出现的次数,则二分查找就派上了用场。
//核心:二分查找
public class Solution {
public int GetNumberOfK(int [] array , int k) {
if(array==null && array.length==0){
return 0;
}
int firstK=getFirstK(array,0,array.length-1,k);
int lastK = getLastK(array,0,array.length-1,k);
if(firstK != -1 && lastK != -1){
return lastK - firstK + 1;
}
return 0;
}
//二分查找,找到第一次出现K的位置
int getFirstK(int[] array,int low,int high,int k){
if(low>high){
return -1;
}
int mid = (low+high)>>1;
if(array[mid]<k){
return getFirstK(array,mid+1,high,k);
}else if(array[mid]>k){
return getFirstK(array,low,mid-1,k);
}else if((mid-1) >=0 && array[mid-1] == k){
//往小的方向找
return getFirstK(array,low,mid-1,k);
}
return mid;
}
//循环查找,找到最后一次出现K的位置
int getLastK(int[] array,int low,int high,int k){
int length = array.length;
if(low>high){
return -1;
}
int mid = (low+high)>>1;
if(array[mid]<k){
return getLastK(array,mid+1,high,k);
}else if(array[mid]>k){
return getLastK(array,low,mid-1,k);
}else if((mid+1) <length && array[mid+1] == k){
//往大的方向找
return getLastK(array,mid+1,high,k);
}
return mid;
}
}