https://www.nowcoder.com/questionTerminal/70610bf967994b22bb1c26f9ae901fa2
来源:牛客网
【题目】
统计一个数字在排序数组中出现的次数。
【代码】
//面试就到这里,你可以回去了
public int GetNumberOfK(int [] arr , int k) {
if(arr==null||arr.length==0) return 0;
int count=0;
for(int i=0;i<arr.length;i++){
if(arr[i]==k){
count++;
}
}
return count;
}
//看到有序想二分查找
public int GetNumberOfK(int [] array , int k) {
int length = array.length;
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(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(int [] array , int k, int start, int end){
int length = array.length;
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;
}
新方法
因为data中都是整数,所以可以稍微变一下,不是搜索k的两个位置,而是搜索k-0.5和k+0.5
这两个数应该插入的位置,然后相减即可。
public int GetNumberOfK(int [] arr , int k) {
int length = arr.length;
return find(arr,k+0.5)-find(arr,k-0.5);
}
public int find(int[] arr,double k){
int i=0;
int j=arr.length-1;
while(i<=j){
int mid=i+(j-i)/2;
if(arr[mid]<k) i=mid+1;
else if(arr[mid]>k) j=mid-1;
}
return i;
}