最简单的二分查找算法
该部分算法只需要查找出数组中值等于value的元素,数组中不存在重复的元素,因此是最简单的一类二分查找,后续会在此基础上进行拓展。
public int bsearch(int[] a, int n, int value) {
int low = 0;
int high = n - 1;
while (low <= high) {
int mid = (low + high) / 2;
if (a[mid] == value) {
return mid;
} else if (a[mid] < value) {
low = mid + 1;
} else {
high = mid - 1;
}
}
return -1;
}
low、high、mid都是待查找的数组下标,分别指向查找区间的初始位置、结束位置、中间位置。每次都会根据a[mid]的值更新下一次查找的区间范围。需要注意的地方如下:
- 循环退出条件
每次的循环退出条件应当是初始位置超过了结束位置,因此while循环的条件是low <= high而非low < high - mid的取值
为了规避数据溢出的问题,计算mid的值时不能使用mid=(low+high)/2,而是应该使用mid = low + (high - low) / 2,在此基础上又可以转化为位运算,写作mid = low + ((high - low) >> 1),对性能进行优化。 - low和high的更新
low = mid + 1,high = mid - 1,不能直接将low或high更新为mid,不仅没有意义而且可能会导致死循环。如low = high = 3,如果此时a[3] != value,程序将处于死循环状态。
以上只是二分查找算法最简单的一类应用,只需要查找等于给定值的元素即可,除此之外还有查找第一个元素、查找最后一个元素等变体,下面列出这些变体的题解,是我看过的最容易理解的代码。除此之外还有一些简化但是更难理解的写法,后续有时间会放出来作为对照。
查找第一个等于给定值的二分查找算法
public int bsearch(int[] a, int n, int value) {
int low = 0;
int high = n - 1;
while (low <= high) {
int mid = low + ((high - low) >> 1);
if (a[mid] > value) {
high = mid - 1;
} else if (a[mid] < value) {
low = mid + 1;
} else {
if ((mid == 0) || (a[mid - 1] != value)) return mid;
else high = mid - 1;
}
}
return -1;
}
查找最后一个等于给定值的二分查找算法
public int bsearch(int[] a, int n, int value) {
int low = 0;
int high = n - 1;
while (low <= high) {
int mid = low + ((high - low) >> 1);
if (a[mid] > value) {
high = mid - 1;
} else if (a[mid] < value) {
low = mid + 1;
} else {
if ((mid == n - 1) || (a[mid + 1] != value)) return mid;
else low = mid + 1;
}
}
return -1;
}
查找第一个大于等于给定值的二分查找算法
public int bsearch(int[] a, int n, int value) {
int low = 0;
int high = n - 1;
while (low <= high) {
int mid = low + ((high - low) >> 1);
if (a[mid] >= value) {
if ((mid == 0) || (a[mid - 1] < value)) return mid;
else high = mid - 1;
} else {
low = mid + 1;
}
}
return -1;
}
查找最后一个小于等于给定值的二分查找算法
public int bsearch(int[] a, int n, int value) {
int low = 0;
int high = n - 1;
while (low <= high) {
int mid = low + ((high - low) >> 1);
if (a[mid] > value) {
high = mid - 1;
} else {
if ((mid == n - 1) || (a[mid + 1] > value)) return mid;
else low = mid + 1;
}
}
return -1;
}