剑指Offer解题报告(Java版)——排序数组中某个数的个数 38

 
 

分析问题

 
 

问题只需要找到排序数组中某个数K的个数，由于已经是排序了，K一定是在一堆的，所以我们只需要找到第一个K的index1，然后找到最后一个K的index2就可以了

 
 

而寻找K的过程我们一般通过二分法查找，这样时间复杂度能降到logn

 
 

解决问题

 
 

我们通过二分法寻找k，如果中间的数小于k，那么在前半段找k；如果中间的数大于k，那么在后半段找k，那么如何判断找到的k是否是一堆k的边界呢

 
 

找第一个k的时候判断方式如下：

如果中间的数等于k，那么先判断前一个数是否存在，如果存在且等于k，在前半段找

如果存在不等于k或者不存在，那么这个数就是第一个k

 
 

找最后一个k的时候判断方式如下：

如果中间的数等于k，那么先判断后一个数是否存在，如果存在且等于k，在后半段找，如果存在不等于k或者不存在，那么这个数就是最后一个k

 
 

主要代码

 
 

int getLastK(int[] data,int length,int k,int start,int end){

 
 

if (start>end) {

return -1;

}

 
 

int midIndex=(start+end)/2;

int midData=data[midIndex];

if (data[midIndex]==k) {

if( (midIndex<length-1&&data[midIndex+1]!=k)||midIndex==length-1) {

return midIndex;

}else {

start=midIndex+1;

}

}else if (midData<k) {

start=midIndex+1;

}else {

end=midIndex-1;

}

return getLastK(data,length,k,start,end);

 
 

}

 
 

int getFirstK(int[] data,int length,int k,int start,int end){

if (start>end) {

return -1;

}

 
 

int midIndex=(start+end)/2;

int midData=data[midIndex];

if (data[midIndex]==k) {

if( (midIndex>0&&data[midIndex-1]!=k)||midIndex==0) {

return midIndex;

}else {

end=midIndex-1;

}

}else if (midData>k) {

end=midIndex-1;

}else {

start=midIndex+1;

}

return getFirstK(data,length,k,start,end);

 
 

}

 
 

int getNumOfK(int[] data,int length,int k){

int result = 0;

if (data!=null&&length>0) {

int first=getFirstK(data, length, k, 0, length-1);

int last=getLastK(data, length, k, 0, length-1);

System.out.println(first);

System.out.println(last);

if (first>-1&&last>-1) {

result=last-first+1;

}

}

return result;

}