二分法模板

二分法是一个很基本的算法,给定一个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;
}
  • 3
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值