二分查找是一种在每次比较之后将查找空间一分为二的算法。每次需要查找集合中的索引或元素时,都应该考虑二分查找。如果集合是无序的,我们可以总是在应用二分查找之前先对其进行排序。
成功的二分查找的 3 个部分
二分查找一般由三个主要部分组成:
- 预处理 —— 如果集合未排序,则进行排序。
- 二分查找 —— 使用循环或递归在每次比较后将查找空间划分为两半。
- 后处理 —— 在剩余空间中确定可行的候选者。
模板
1
int binarySearch(vector<int>& nums, int target){
if(nums.size() == 0)
return -1;
int left = 0, right = nums.size() - 1;
while(left <= right){
// Prevent (left + right) overflow
int mid = left + (right - left) / 2;
if(nums[mid] == target){ return mid; }
else if(nums[mid] < target) { left = mid + 1; }
else { right = mid - 1; }
}
// End Condition: left > right
return -1;
}
初始条件:left = 0, right = length-1
终止:left > right
向左查找:right = mid-1
向右查找:left = mid+1
2
int binarySearch(vector<int>& nums, int target){
if(nums.size() == 0)
return -1;
int left = 0, right = nums.size();
while(left < right){
// Prevent (left + right) overflow
int mid = left + (right - left) / 2;
if(nums[mid] == target){ return mid; }
else if(nums[mid] < target) { left = mid + 1; }
else { right = mid; }
}
// Post-processing:
// End Condition: left == right
if(left != nums.size() && nums[left] == target) return left;
return -1;
}
初始条件:left = 0, right = length
终止:left == right
向左查找:right = mid
向右查找:left = mid+1
这个模板可以防止越界.
例题:
我们正在玩一个猜数字游戏。 游戏规则如下:
我从 1 到 n 选择一个数字。 你需要猜我选择了哪个数字。
每次你猜错了,我会告诉你这个数字是大了还是小了。
你调用一个预先定义好的接口 guess(int num)
,它会返回 3 个可能的结果(-1
,1
或 0
):
-1 : 我的数字比较小
1 : 我的数字比较大
0 : 恭喜!你猜对了!
示例 :
输入: n = 10, pick = 6
输出: 6
class Solution {
public:
int guessNumber(int n) {
int L=1,R=n;
while(L<=R)
{
int mid=L+(R-L)/2;//L+R会溢出,超出int
cout<<mid;
int tar=guess(mid);
if(tar==0) return mid;
else if(tar==1) L=mid+1;
else if(tar==-1) R=mid-1;
}
return -1;
}
};
二分查找的另外一个变种借用了递归
- 寻找峰值
峰值元素是指其值大于左右相邻值的元素。
给定一个输入数组 nums,其中 nums[i] ≠ nums[i+1],找到峰值元素并返回其索引。
数组可能包含多个峰值,在这种情况下,返回任何一个峰值所在位置即可。
你可以假设 nums[-1] = nums[n] = -∞。
示例 1:
输入: nums = [1,2,3,1]
输出: 2
解释: 3 是峰值元素,你的函数应该返回其索引 2。
示例 2:
输入: nums = [1,2,1,3,5,6,4]
输出: 1 或 5
解释: 你的函数可以返回索引 1,其峰值元素为 2;
或者返回索引 5, 其峰值元素为 6。
说明:
你的解法应该是 O(logN) 时间复杂度的。
class Solution {
public:
int findPeakElement(vector<int>& nums) {
if(nums.size()==1) return 0;
return findp(nums,0,nums.size()-1);
}
int findp(vector<int>&nums,int left,int right)
{
if(left>right) return -1; //注意这里是if 不是while 所以这里一直都是闭区间或者说一直都是一个点
int mid=left+(right-left)/2;
int l=findp(nums,left,mid-1);
if(l!=-1)
{
return l;
}
else if(mid==0)
{
if(nums[mid]>nums[mid+1])
return mid;
}
else if(mid==nums.size()-1)
{
if(nums[mid]>nums[mid-1]);
return mid;
}
else if(nums[mid]>nums[mid-1]&&nums[mid]>nums[mid+1])
{
return mid;
}
int r=findp(nums,mid+1,right);
if(r!=-1)
{
return r;
}
// else if(mid==nums.size()-1)
// {
// if(nums[mid]>nums[mid-1]);
// return mid;
// }
// else if(nums[mid]>nums[mid-1]&&nums[mid]>nums[mid+1])
// {
// return mid;
// }
return -1;
}
};
递归需要清楚的点
(1)终止条件以及函数三要素
(2)这一层做了什么
(3)下一层返回上一层什么
(4)相信这个函数能做到
(5)用其中一层验证
面试题 08.03. 魔术索引
魔术索引。 在数组A[0...n-1]中,有所谓的魔术索引,满足条件A[i] = i。给定一个有序整数数组,编写一种方法找出魔术索引,若有的话,在数组A中找出一个魔术索引,如果没有,则返回-1。若有多个魔术索引,返回索引值最小的一个。
示例1:
输入:nums = [0, 2, 3, 4, 5]
输出:0
说明: 0下标的元素为0
示例2:
输入:nums = [1, 1, 1]
输出:1
说明:
nums长度在[1, 1000000]之间
此题为原书中的 Follow-up,即数组中可能包含重复元素的版本
class Solution {
public:
int findMagicIndex(vector<int>& nums) {
//二分查找
return findmargic(nums,0,nums.size()-1);
}
int findmargic(vector<int>& nums,int left,int right)
{
if(left>right) return -1; //这里维持着一个【left,right)区间 注意if与while的区别
int mid=left+(right-left)/2;
int tmp_a=findmargic(nums,left,mid-1);
if(tmp_a!=-1)
{
return tmp_a;
}
else if(nums[mid]==mid)
{
return mid;
}
return findmargic(nums,mid+1,right);
}
};