二分查找(Binary Search)
二分查找是从有序数组中的中间位置开始查找特定元素的算法,又称折半查找。
算法流程:
- 从有序数组的中间位置开始与给定元素进行比较。
- 若二者相等,则返回下标。
- 若不相等,则在大于或小于给定元素的那一半数组中重复以上步骤。
- 若在某一步数组为空,表明查找失败。
每次比较,数组长度n减小一半,所以二分查找的时间复杂度为O(logn)。
1.基本二分查找
问题描述:从有序数组中查找value,存在返回下标,不存在返回-1。
1.1 非递归
int BinarySearch(vector<int> array, int value){
if(array.empty())
return -1;
int left = 0;
int right = array.size()-1;
while(left <= right)//当数组只有一个元素 左右相等
{
int middle = left + (right - left>>1);//加法可能溢出,同时若是迭代器也不能相加
if(array[midlle] > value){
right = middle - 1;
}
else if (array[midlle] < value){
left = middle + 1;
}
else
return middle;
}
return -1;
}
1.2 递归
int binarySearch(vector<int>& nums, int target, int l, int r){
if(l>r) return -1;
int mid = l + (r-l)/2;
if(nums[mid] < target) return binarySearch(nums, target, mid+1, r);
if(numd[mid] > target) return binarySearch(nums, target, l , mid - 1);
return mid;
}
2. 变形
问题描述:从给定有序数组中查找第一个等于value的元素,查找成功返回下标,查找失败返回-1。
类似问题:
1.查找第一个不小于value的元素。
2.查找第一个大于value的元素。
即:STL中的lower_bound()和uppper_bound()。
找到了等于(大于)value的元素,记录这个下标,继续到相应的折半数组中查找,直到数组为空。
int BinarySearch1(vector<int> array, int value){
if(array.empty())
return -1;
int left = 0;
int right = array.size()-1;
while(left <= right){
unsigned int middle = left + (right - left >> 1);
if(array[middle] >= value)//只要满足条件 right指示middle左侧元素
right = middle - 1;
else
left = middle + 1;
}
return array[right + 1] == value ? right + 1:-1;
}
- 查找第一个大于value的元素,将 >= 改为> ,将== 改为 >
- 查找最后一个等于value的元素 ,将>=改为>,将right+1改为left-1
看array[middle]与value相等时往右还是往左。