算法:数的范围

题目:

Solution:

#include <iostream>
using namespace std;

const int N = 1e5 + 10;

int n, q;
int a[N];

int find(int k, int type) //type为0表示查找起始位置,type为1表示查找终止位置
{
    int l = 0, r = n - 1, ans = -1;
    while (l <= r)
    {
        int mid = l + r >> 1;
        if (a[mid] >= k)
        {
            if (a[mid] == k)
                ans = mid;
            r = mid - 1;
        }
        else
            l = mid + 1;
    }
    if (type == 0)
        return ans;
    l = 0, r = n - 1, ans = -1;
    while (l <= r)
    {
        int mid = l + r >> 1;
        if (a[mid] <= k)
        {
            if (a[mid] == k)
                ans = mid;
            l = mid + 1;
        }
        else
            r = mid - 1;
    }
    return ans;
}

int main()
{
    cin >> n >> q;
    for (int i = 0; i < n; i ++ )
        cin >> a[i];
    while (q -- )
    {
        int k;
        cin >> k;
        int l = find(k, 0), r = find(k, 1);
        cout << l << ' ' << r << endl;
    }
    return 0;
}

二分法查找法。

二分法是一种高效的查找算法,适用于有序数组。

在该代码中,find函数使用了两次二分查找。第一次查找是为了找到元素k的起始位置,通过不断更新左右边界来逼近目标位置。如果当前中间位置的元素大于等于k,则将右边界更新为mid-1,否则将左边界更新为mid+1。当找到元素k时,将结果记录在ans变量中。

第二次查找是为了找到元素k的终止位置。同样通过不断更新左右边界来逼近目标位置。如果当前中间位置的元素小于等于k,则将左边界更新为mid+1,否则将右边界更新为mid-1。当找到元素k时,将结果记录在ans变量中。

最后,将起始位置和终止位置分别作为函数的返回值,输出到标准输出流中。

这样就可以实现在升序排列的数组中使用二分法查找指定元素的起始位置和终止位置。

二分查找的时间复杂度为O(logn),其中n为数组的长度。通过二分法,可以快速找到目标元素在数组中的位置。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值