笔者今天无意间看到了之前收藏的一个公众号推文,内容是关于二分查找的。笔者在这里就和大家絮叨絮叨。说起二分查找,想必大家应该比较熟悉吧,Java中Arrays中的binarySearch()
函数表示的正是二分查找算法。该算法主要用于在有序数组中查找指定元素key,若元素存在,则直接返回下标。这里笔者在LeetCode上找了一个二分查找的直接例子——传送门。二分查找代码如下
public int search(int[] nums, int target) {
int start = 0, end = nums.length - 1;
while (start <= end) {
int mid = start + (end - start) / 2;
if (nums[mid] > target) {
end = mid - 1;
} else if (nums[mid] < target) {
start = mid + 1;
} else return mid;
}
return -1;
}
以上就是二分查找最基本的用法。笔者这里整理了一下公众号的推文,大致分为以下两种:
变式一
public int search(int[] nums, int key){
int start = 0, end = nums.length - 1;
while (start <= end) {
int mid = start + ((end - start) >> 1);
if (nums[mid] >= key) {
end = mid - 1;
} else {
start = mid + 1;
}
}
/* 设数组为:[1, 2, 2, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 6, 6, 7],key = 5。那么在该二分查找的过程中,start和end
都会趋于数组中元素5开始的地方,直至start>end。此时start指向数组中key理论上(若key不存在,则对应插入的位置)
在nums中的第一个位置,而end=start-1。因此该形式可以返回三种类型的结果,即:
1)第一个等于指定key的元素下标,若不存在,返回-1。要求start<nums.length && nums[start]=key,因key大于nums中
的最大值时start会等于nums.length。
2)第一个大于等于key的元素下标,直接返回start即可。
3)最后一个小于key的元素下标,直接返回end即可。*/
}
变式二
public int search(int[] nums, int key){
int start = 0, end = nums.length - 1;
while (start <= end) {
int mid = start + ((end - start) >> 1);
if (nums[mid] > key) {
end = mid - 1;
} else {
start = mid + 1;
}
}
/* 设数组为:[1, 2, 2, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 6, 6, 7],key = 5。那么在二分查找的过程中,start和end
均会趋于数组中元素5结束的地方,直至start>end。此时start指向数组中(key+1)理论上(若key+1不存在于nums中,则
对应插入的位置)在nums中的第一个位置。而end=start-1.此时同样可以返回三种结果,即:
1)最有一个等于指定key的元素下标,若不存在,返回-1。要求end>=0 && nums[end]=key,因为key小于nums中的最小值
的时候end会变成-1。
2)最后一个小于等于key的元素下标,直接返回end。
3)第一个大于key的元素下标,直接返回start。 */
}