题目:数字在排序数组中出现的次数
题目描述
统计一个数字在排序数组中出现的次数。
解题思路:
由于是排序数组,可以使用二分查找法:查找该数第一次出现的位置和该数最后一次出现的位置定义两个指针,start end来查找该数第一次出现的位置left和该数最后一次出现的位置right
代码实现:
public class Solution {
public static int GetNumberOfK(int [] array , int k) {
//该数第一次出现的位置:
int left=getLeft(array,k);
//该数最后一次出现的位置:
int right=getRight(array,k);
return right-left+1;
}
private static int getRight(int[] array, int k) {
int start=0;
int end=array.length-1;
while(start<=end) {
int mid=start+(end-start)/2;
//如果中间元素小于该数k,说明该数存在于数组的后半段
//如果中间元素和k相等的话呢?因为查找的是该数最后出现的位置,所以可以将其定位在数组的后半段
if(array[mid]<=k) {
start=mid+1;
}else {
end=mid-1;
}
}
return end;
}
private static int getLeft(int[] array,int k) {
int start=0;
int end=array.length-1;
while(start<=end) {
int mid=start+(end-start)/2;
//如果中间元素大于该数k,说明该数存在于数组的前半段
//如果中间元素和k相等的话呢?因为查找的是该数最开始出现的位置,所以可以将其定位在数组的前半段
if(array[mid]>=k) {
end=mid-1;
}else {
start=mid+1;
}
}
return start;
}
}
ps: 评论区有一位大佬给出了查找k+0.5和k-0.5的位置之差的,感觉很新颖,可以借鉴一下。
public class Solution {
public int GetNumberOfK(int [] array , int k) {
//该数第一次出现的位置:
int left=getRes(array,k-0.5);
//该数最后一次出现的位置:
int right=getRes(array,k+0.5);
return right-left;
}
private static int getRes(int[] array, double k) {
int start=0;
int end=array.length-1;
while(start<=end) {
int mid=start+(end-start)/2;
if(array[mid]>k) {
end=mid-1;
}else if(array[mid]<k) {
start=mid+1;
}
}
return start;
}
}