题目:数字在排序数组中出现的次数
统计一个数字在排序数组中出现的次数。例如,输入{1,2,3,3,3,3,4,5}和数字3,由于3在这个数组中出现了4次,因此输出4。
思路:改进运用二分查找
接下来我们思考如何更好地利用二分查找算法。假设我们是统计数字k在排序数组中出现的次数。在前面的算法中时间主要消耗在如何确定重复出现的数字的第一个k和最后一个k的位置上,有没有可能用二分查找算法直接找到第一个k及最后一个k呢?
我们先分析如何用二分查找算法在数组中找到第一个k。二分查找算法总是先拿数组中间的数字和k作比较。如果中间的数字比k大,那么k只有可能出现在数组的前半段,下一轮我们只在数组的前半段查找就可以了。如果中间的数字比k小,那么k只有可能出现在数组的后半段,下一轮我们只在数组的后半段查找就可以了。如果中间的数字和k相等呢?我们先判断这个数字是不是第一个k。如果位于中间数字的前面一个数字不是k,此时中间的数字刚好就是第一个k。如果中间数字的前面一个数字也是k,也就是说第一个k肯定在数组的前半段,下一轮我们仍然需要在数组的前半段查找。
基于以上思路,java参考代码如下:
package chapter6;
public class P263_NumberOfK {//另一种方法
public static int findnumbersofk(int[] data,int target){
int left=0,right=data.length-1;
int[] res=new int[2];
while (left<right){
int mid=left+(right-left)/2;
if(data[mid]<target)
left=mid+1;
else
right=mid;
}
if(data[left]!=target) return 0;
res[0]=left;
right=data.length;
while (left<right){
int mid=left+(right-left)/2;
if(data[mid]<=target)
left=mid+1;
else
right=mid;
}
res[1]=left-1;
return res[1]-res[0]+1;
}
public static int getFirst(int[] arr,int target){//先找到第一个目标数字
int left=0,right=arr.length-1;
while (left<=right){
int mid=left+(right-left)/2;
if(arr[mid]==target){
if(mid>0&&arr[mid-1]!=target||mid==0){
return mid;
}else {
right=mid-1;
}
}else if(arr[mid]>target){
right=mid-1;
}else {
left=mid+1;
}
}
return -1;
}
public static int getLast(int[] arr,int target){//再找到最后一个目标数字
int left=0,right=arr.length-1;
while (left<=right){
int mid=left+(right-left)/2;
if(arr[mid]==target){
if(mid<arr.length-1&&arr[mid+1]!=target||mid==arr.length-1){
return mid;
}else {
left=mid+1;
}
}else if(arr[mid]>target){
right=mid-1;
}else {
left=mid+1;
}
}
return -1;
}
public static int findnumbersofk2(int[] arr,int target){//求得所有目标数字的个数
if(arr==null||arr.length<1) return -1;
int start=getFirst(arr,target);
int end=getLast(arr,target);
if(start==-1||end==-1) return 0;
return end-start+1;
}
public static void main(String[] args){
int target=3;
int[] data={1,2,3,3,3,3,4,5};
int[] data1={1,2,4,5};
int[] data2={3,3,3,3,3,3};
System.out.println(findnumbersofk(data,target));
System.out.println(findnumbersofk(data1,target));
System.out.println(findnumbersofk(data2,target));
System.out.println(findnumbersofk2(data,target));
System.out.println(findnumbersofk2(data1,target));
System.out.println(findnumbersofk(data2,target));
}
}
测试用例:
a.功能测试(数组中包含要查找的数字;;数组中没有要查找的数字;要查找的数字在数组中出现一次或者多次)。
b.边界值测试(查找数组在的最大值、最小值;数组中只有一个数字)。
c.特殊输入测试(表示数组的指针为nullptr指针)。
参考:
https://www.jianshu.com/p/a5bda52fe134
http://www.cnblogs.com/grandyang/p/4409379.html