整数二分模板,彻底解决边界问题
Published on 2022-11-17 23:04 in 暂未分类 with 林动

整数二分模板,彻底解决边界问题

     //区间分为[l,mid]和[mid+1,r],如下,x<=a[mid]的判断条件,使得x要么在[l,mid],要么[mid+1,r]
    //最终l会等于r
         while(l<r)
            {
                int mid=l+r>>1;
                if(a[mid]>=x)r=mid;
                else l=mid+1;
            }
     //区间分为[l,mid-1]和[mid,r],如下,x>=a[mid]的判断条件,使得x要么在[l,mid-1],要么[mid,r]
            while(l<r)
            {
                int mid=l+r+1>>1;
                if(a[mid]<=x)l=mid;//不加1死循环条件
                else r=mid-1;
            }
     
    //当一个单调区间中有连续多个x时候,第一个模板会取到最左边那个x下标,因为x==a[mid]时候是边界向左压缩。同理,第二个取到最右边的x下标
     
    //第二个模板算mid要+1因为区间长度为2时,mid算出来等于l,而第二个模板存在死循环条件:mid给l赋值。
    
    
    2022.3.6
    //第一个会取到最左边那个符合check条件的下标
    //第二个会取到最右边那个符合check条件的下标
    

    例题

    链接:http://noi.openjudge.cn/ch0111/01/

    01:查找最接近的元素
    总时间限制: 1000ms 内存限制: 65536kB
    描述
    在一个非降序列中,查找与给定值最接近的元素。

    输入
    第一行包含一个整数n,为非降序列长度。1 <= n <= 100000。
    第二行包含n个整数,为非降序列各元素。所有元素的大小均在0-1,000,000,000之间。
    第三行包含一个整数m,为要询问的给定值个数。1 <= m <= 10000。
    接下来m行,每行一个整数,为要询问最接近元素的给定值。所有给定值的大小均在0-1,000,000,000之间。
    输出
    m行,每行一个整数,为最接近相应给定值的元素值,保持输入顺序。若有多个值满足条件,输出最小的一个。
    样例输入
    3
    2 5 8
    2
    10
    5
    样例输出
    8
    5

    AC代码:

    #include <iostream>
    
    using namespace std;
    
    const int N=1e5+5;
    
    int n,a[N],m,x,l,r,i;
    
    bool check(int u)
    {
    	//下面两种判断条件都可以
    	//if(a[u]>=x||a[u]<x&& (x-a[u])<=(a[u+1]-x))return true;
    	//return false;
    	if(a[u]<x&&(x-a[u])>(a[u+1]-x))return false;
    	return true;
    }
    
    int main()
    {
        cin>>n;
        for(i=0;i<n;++i)cin>>a[i];
        cin>>m;
        while(m--)
        {
        	cin>>x;
        	l=0,r=n-1;
        	//二分就是考虑什么时候向左压缩什么时候向右压缩 
        	while(l<r)
        	{
        		int mid=l+r>>1;//因为mid是下取整,所以mid 永远不会取到初始的右边界 
        		//同理,第二个模板永远不会取到初始的左边界
        		if(check(mid))r=mid;//满足条件就向左边压缩 
        		else l=mid+1;//向右边压缩 
        	}
        	cout<<a[l]<<endl;
        }
        return 0;
    }
    

    还有一种二分的方式,例题:leetcode 33. 搜索旋转排序数组 中等 二分

    class Solution {
        public int search(int[] nums, int target) {
        	int n=nums.length;
        	int l=0,r=n-1;
        	while(l<=r){
        		int mid=l+r>>1;
        		if(nums[mid]==target)return mid;
        		if(nums[l]<=nums[mid]){
        			if(target>=nums[l]&&target<nums[mid])r=mid-1;
        			else l=mid+1;
        		}else{
        			if(target>nums[mid]&&target<=nums[r])l=mid+1;
        			else r=mid-1;
        		}
        	}
        	return -1;
        }
    }
    
    while(l<=r)
    {
    	int mid=l+r>>1;
    	if(a[mid]==x)return mid;
    	if(a[mid]>x)r=mid-1;
    	else l=mid+1;
    }
    跳出循环后l==r
    
    posted @   林动  阅读(18)  评论(0编辑  收藏  举报
    相关博文:
    阅读排行:
    · TypeScript + Deepseek 打造卜卦网站:技术与玄学的结合
    · Manus的开源复刻OpenManus初探
    · AI 智能体引爆开源社区「GitHub 热点速览」
    · 三行代码完成国际化适配,妙~啊~
    · .NET Core 中如何实现缓存的预热?
    点击右上角即可分享
    微信分享提示