题目:
统计一个数字在排序数组中出现的次数. 例如输入排序数组{1,2,3,3,3,3,4,5},由于3在这个数中出现了4次,输出4.
思路:
如何利用二分查找找到第一个k? 二分查找算法总是先拿数组中间的数组和k作比较.
如果中间数字比k大,那么k只可能出现数组的前半段,下一轮只在数组的前半段查找就可以了.
如果中间数字比k小,那么k只可能出现数组的后半段,下一轮只在数组的后半段查找就可以了.
如果中间数组和k相等,先判断这个数字是不是第一个k.
如果位于中间数字前面的一个数字不是k,那么此时中间的数字刚好就是第一个k.
如果中间数字的前面一个数字也是k,也就是说第一个k肯定在数组的前半段,下一轮仍然需要在数组的前半段查找.
基于同样的思路可以在排序数组中找到最后一个k.
如果中间数字比k大,那么k只能出现在数组的前半段.
如果中间数字比k小,那么k只能出现在数组的后半段.
如果中间数子和k相等,需要判断这个数字是不是最后一个k.
如果位于中间数字后面一个数字不是k,那么此时中间的数字刚好就是最后一个k.
如果中间数字的后面一个数字也是k,也就是说第一个k肯定在数组的后半段,下一轮仍然需要在数组的后半段查找.
public static void main(String[] args) {
int[] a = {1, 2, 3, 3, 3, 3, 4, 5};
System.out.println(getKCount(a, 3));
}
static int getKCount(int[] a, int k) {
if (a != null && a.length != 0) {
int first = getFirstK(a, k, 0, a.length - 1);
int last = getLastK(a, k, 0, a.length - 1);
if (first > -1 && last > -1) {
return last - first + 1;
}
}
return -1;
}
static int getFirstK(int[] a, int k, int start, int end) {
if (start > end) {
return -1;
}
int mid = (start + end) / 2;
int midData = a[mid];
if (midData == k) {
if ((mid > 0 && a[mid - 1] != k) || mid == 0) {
return mid;
} else {
end = mid - 1;
}
} else if (midData < k) {
// 在后半段
start = mid + 1;
} else if (midData > k) {
// 在前半段
end = mid - 1;
}
return getFirstK(a, k, start, end);
}
static int getLastK(int[] a, int k, int start, int end) {
if (start > end) {
return -1;
}
int mid = (start + end) / 2;
int midData = a[mid];
if (midData == k) {
if ((mid > 0 && a[mid + 1] != k) || mid == a.length - 1) {
return mid;
} else {
start = mid + 1;
}
} else if (midData < k) {
// 在后半段
start = mid + 1;
} else if (midData > k) {
// 在前半段
end = mid - 1;
}
return getLastK(a, k, start, end);
}