题目描述
统计一个数字在排序数组中出现的次数。
如果数组中不包含该数字,返回-1。
题解思路:
数组是排序的,并且是查找,应该想到二分查找。
但是并不是找那个数出现的范围,因为如果整个数组都是同一个数的话复杂度就是O(n)
求第一个K:
(1)A[mid]==K:说明第一个K在左边,开始比较A[mid-1]
①mid刚好是0,返回mid
②A[mid-1]!=k,返回mid。
③A[mid-1]==k,mid-1>0,则right = mid-1
(2)A[mid]>K:说明第一个K在左边
right = mid - 1;
(3)A[mid]<K:说明第一个K在右边
left = mid + 1;
求最后一个K:
(1)A[mid] = K:说明最后一个K在右边,开始比较A[mid+1]
①mid==n-1,返回mid
②A[mid+1]!=k,返回mid
③A[mid+1]==k,left = mid + 1
public class Solution {
public int GetNumberOfK(int [] array , int k) {
int k1 = getFirstK(array,k);
int k2 = getEndK(array,k);
if(k1==-1||k2==-1){
return 0;
}
return k2 - k1 + 1;
}
public int getEndK(int [] array,int k){//寻找最后一个K
int n = array.length;
int left = 0,right = n-1;
while(left<=right){
int mid = (left + right)/2;
if(array[mid]==k){
if((mid==n-1)||(array[mid+1]!=k)){
return mid;
}
else if(array[mid+1]==k){
left = mid + 1;
}
}
else if(array[mid]>k){//K在mid的左边
right = mid - 1;
}
else if(array[mid]<k){//K在mid的右边
left = mid + 1;
}
}
return -1;
}
public int getFirstK(int [] array,int k){//寻找第一个K
int n = array.length;
int left = 0,right = n-1;
while(left<=right){
int mid = (left + right)/2;
if(array[mid]==k){//第一个k在mid左边
if((mid==0)||(array[mid-1]!=k)){
return mid;
}
else if(array[mid-1]==k){
right = mid - 1;
}
}
else if(array[mid]>k){//第一个k在mid左边
right = mid - 1;
}
else{//第一个k在mid右边
left = mid + 1;
}
}
return -1;
}
}