基础算法:整数二分——数的范围

 题目:

给定一个按照升序排列的长度为 n 的整数数组,以及 q 个查询。

对于每个查询,返回一个元素 k 的起始位置和终止位置(位置从 0 开始计数)。

如果数组中不存在该元素,则返回 -1 -1

输入格式

第一行包含整数 n 和 q,表示数组长度和询问个数。

第二行包含 n 个整数(均在 1∼100001∼10000 范围内),表示完整数组。

接下来 q 行,每行包含一个整数 k,表示一个询问元素。

输出格式

共 q 行,每行包含两个整数,表示所求元素的起始位置和终止位置。

如果数组中不存在该元素,则返回 -1 -1

输入样例:
6 3
1 2 2 3 3 4
3
4
5
输出样例:
3 4
5 5
-1 -1

/*
    整数二分查找算法
    
    步骤:
    1.找中间值mid = l + r >> 1;
    2.根据题目要求更新边界,每次缩小一半,
    找答案所在的区间,当区间长度为1时就是答案
    3.根据模板的更新方式来选择是否+1
        如果更新是l = mid,就mid = l + r + 1 >> 1;c++中是向下取整,不然会死循环
        如果更新是r = mid,就不用
    本题已给出有序数组,需要写两个二分,
    一个需要找到>=x的第一个数,当查找结束时,l与r相遇,l所在元素若是x则一定是x出现最小位置,因为l左边的元素必然都小于x
    另一个需要找到<=x的最后一个数,同理。
*/

#include <iostream>
using namespace std;

const int N = 1e5;

int n,m;
//n:数组长度,m:要查找的数量

int q[N];

int main()
{
    cin >> n >> m;
    for(int i = 0; i < n; i++) scanf("%d", &q[i]);
    
    while(m--)
    {
        //x:要查找的数
        int x;
        scanf("%d", &x);
        
        int l = 0, r = n - 1;
        
        //1.找x的起始位置
        while(l < r)
        {
            int mid = l + r >> 1;//先不管加1,看后面的更新条件判断
            
            if(q[mid] >= x) r = mid;
            else l = mid + 1;
        }//结束循环,l==r
        
        if(q[l] != x) cout << "-1 -1" << endl;
        else
        {
            //x的起始位置
            cout << l << ' ';
            
            //2.找x的终止位置
            int l = 0, r = n - 1;
            while(l < r)
            {
                int mid = l + r + 1>> 1;
                if(q[mid] <= x) l = mid;//如果是l = mid,前面的mid = l + r + 1 >>1,不然会死循环
                else r = mid - 1;
            }
            cout << l << endl;
        }
    }
    
    return 0;
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值