数字在排序数组中出现的次数
题目描述:统计一个数字在排序数组中出现的次数。例如,输入排序数组{1,2,3,3,3,3,4,5}和数字3,由于数字3在该数组中出现了4次,所以函数返回4。
-
解题思路:
二分查找
既然输入的数组是有序的,所以我们就能很自然的想到用二分查找算法。以题目中给的数组为例,一个比较自然的想法是用二分查找先找到一个3,由于要计算的是输出的次数,所以需要在找到的这个3的左右两边分别再进行顺序扫描,进而得到3的个数,这样最坏的情况下时间复杂度仍然是O(n),和直接顺序扫描的效率相同。
因此,需要考虑怎样更好的利用二分查找算法,由于数组有序,如果知道了第一个k出现的位置和最后一个k出现的位置,那么我们就可以直接算出有多少个k。因此将思路转化为通过二分查找求第一个和最后一个k出现的位置。
以第一个k出现的位置为例,利用二分查找算法可以直接对数组进行二分,而每次总是拿中间的数字和k做比较,如果中间的数字大于k,那么第一个k只有可能出现在左边,下一次直接在数组左半段继续进行二分查找;如果中间的数字小于k,则第一个k只有可能出现在右边,则在右半段再查找;如果中间的数字等于k,我们先判断它前面的一个数字是不是k,如果不是,那么这个中间的数字就是第一个出现的位置,反之,如果中间数字前面的数字是k,那么第一个k仍然在前半段,继续查找。
同理,找最后一个k出现的位置方法类似,可以使用两个函数分别获得。
public class Title37 {
private int findFirstPosition(int[] array, int k) {
int l = 0;
int r = array.length - 1;
while (l < r) {
int mid = (l + r) >> 1;
if (array[mid] == k) {
if(mid - 1 >= 0 && array[mid - 1] == k) {
// 说明mid当前的位置不是初始位置,k的初始位置是在l~mid-1区间
r = mid - 1;
} else {
// 就可以说明mid位置的数字就是k的初始位置
return mid;
}
} else if (array[mid] > k) {
r = mid - 1; // k是属于l~mid-1区间
} else {
l = mid + 1; // k是属于mid+1~r区间
}
}
return l;
}
private int findLastPosition(int[] array, int k) {
int l = 0;
int r = array.length - 1;
while (l < r) {
int mid = (l + r) >> 1;
if (array[mid] == k) {
if(mid + 1 < array.length && array[mid + 1] == k) {
// 说明mid当前的位置不是终止位置,k的初始位置是在mid+1~r区间
l = mid + 1;
} else {
// 就可以说明mid位置的数字就是k的终止位置
return mid;
}
} else if (array[mid] > k) {
r = mid - 1; // k是属于l~mid-1区间
} else {
l = mid + 1; // k是属于mid+1~r区间
}
}
return l;
}
public int GetNumberOfK(int [] array , int k) {
if (array.length == 0) {
return 0;
}
int firstPosition = findFirstPosition(array, k);
int lastPosition = findLastPosition(array, k);
if (array[firstPosition] != k) {
return 0;
}
return lastPosition - firstPosition + 1;
}
}