二分法是一个很基本的算法,给定一个n个有序大的整型数组nums,和一个目标值 target,写一个函数搜索nums种的target,如果目标存在返回下标,如果不存在返回-1。
leetcode中的35题和704题都是二分查找。
nums中第一个大于等于target的元素
我们寻找
int searchInsert(vector<int>& nums, int target)
{
int a = 0, b = nums.size() -1;
while(a <= b)
{
int mid = (a+b)>>1;
if( nums[mid] < target )
{
a = mid+1; //a 的左边的值 < target
}
else
{
b = mid -1;//b 的右边的值 >=target
}
}
//当循环结束的时候 a >b
//a 满足 b 的右边的值的条件 即 nums[a] >= target
return a;
}
1 对于a的左边的元素 i ,0 <=i < a, 有nums[i] <= target
2 对于任意的 j , b < j <= nums.size()-1 有nums[j] > target
3 每次循环的时候都要保证: a减小或者b增大。
当循环结束的时候有 a>b 所以a在b的右边 ,满足b的右边条件,即 nums[a] > target。
nums中第一个大于target的元素
根据上一步的分析, 我们需要构造b的右边元素是大于target的,而mid就是b的右边元素。
int searchInsert( vector<int>& nums, int target)
{
int a = 0, b = nums.size()-1;
while(a<=b)
{
int mid = (a+b)>>1;
if(nums[mid] <= target)
{
}
else
{
b = mid-1;// b的右侧元素大于等于target
}
}
return a;
}
nums中第一个小于等于target的元素
此时nums为降序,需要构造b的右侧小于等于 target。
int searchInsert(vector<int>& nums, int target)
{
int a = 0, b = nums.size();
while(a<=b)
{
int mid = (a+b)>>1;
if(nums[mid] >target )
{
a = mid+1;
}
else
{
b = mid -1; //nums[mid] <= target
}
}
return a;
}
nums中第一个小于target的元素
关键在于构造b的右边都小于target
int searchInsert(vector<int>& nums, int target)
{
int a = 0, b = nums.size();
while(a<=b)
{
int mid = (a+b)>>1;
if(nums[mid] >=target )
{
a = mid+1;
}
else
{
b = mid -1; //nums[mid] <= target
}
}
return a;
}