AcWing 789. 数的范围(二分查找)

题目描述:

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

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

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

输入格式:

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

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

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

输出格式:

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

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

数据范围

1≤n≤100000
1≤q≤10000
1≤k≤10000

整数二分
 有单调性一定可以二分 但没有单调性可以二分
 二分的本质不是单调性,而是边界

 第一种二分

 先找一个mid = (l+r+1) /2
 check mid  true   [mid,r]    l = mid
 false  [l,mid-1]  r = mid-1

如何考虑用什么模板 先写一个check函数  再考虑如何更新区间
    第二种二分
    mid  = (l+r+1) /2
    check mid  true       [l,mid-1]    r = mid
                        false     [mid+1,r]   l = mid+1

题解:根据题意为二分查找,顺序查找肯定会超时。首先本代码分为两个二分查找的函数,其中half_search1返回最先出现该数的位置,half_search2返回最后出现该数的位置。如果不存在返回后只需加一个判断即可。

关于half_search1:

由于返回最先出现该数的位置,当中间位置的数满足条件(a[mid]>=k)时,就从数组的左边找,即r=mid,否则从左边找,即l=mid+1。此时mid=(l+r)/2。

关于half_search2:

由于返回最后出现该数的位置,当中间位置的数满足条件(a[mid]<=k)时,就从数组的右边找,即l=mid,否则从左边找,即r=mid-1。此时mid=(l+r+1)/2。(注此处为l+r+1,如为l+r会陷入死循环)

#include <bits/stdc++.h>
#define ios ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
#define ll long long
#define endl '\n'
#define N 100005
using namespace std;
const ll inf=0x3f3f3f3f;
const double PI=acos(-1);
const ll mod=1e9+7;
ll gcd(ll a, ll b) { return b ? gcd(b, a % b) : a; }
ll lcm(ll a, ll b) { return a * b / gcd(a, b); }
int a[N],x,k;
int half_search1(int l,int r)
{
    while(l<r)
    {
        int mid=(l+r)>>1;
        if(a[mid]>=k) r=mid;
        else l=mid+1;
    }
    return l;
}
int half_search2(int l,int r)
{
    while(l<r)
    {
        int mid=(l+r+1)>>1;
        if(a[mid]<=k) l=mid;
        else r=mid-1;
    }
    return l;
}
int main()
{ios
    int n,q;
    cin >>n>>q;
    for(int i=0;i<n;i++)
    {
        cin >>a[i];
    }
    while(q--)
    {
        cin >>k;
        x=half_search1(0,n-1);
        if(a[x]!=k)
        {
            cout <<"-1 -1"<<endl;
        }else{
            cout <<x<<" ";
            x=half_search2(0,n-1);
            cout <<x<<endl;
        }
    }
    return 0;
}
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
题目链接:https://www.acwing.com/problem/content/4948/ 题目描述: 给定一棵有 $n$ 个结点的树,结点从 $1$ 到 $n$ 编号,每个结点都有一个权值 $w_i$,现在有 $m$ 次操作,每次操作是将树中编号为 $x$ 的结点的权值加上 $y$,然后询问一些节点是否为叶子节点,如果是输出 $1$,否则输出 $0$。 输入格式: 第一行包含两个整 $n$ 和 $m$。 第二行包含 $n$ 个整,其中第 $i$ 个整表示结点 $i$ 的初始权值 $w_i$。 接下来 $n-1$ 行,每行包含两个整 $a$ 和 $b$,表示点 $a$ 和点 $b$ 之间有一条无向边。 接下来 $m$ 行,每行描述一次操作,格式为三个整 $t,x,y$。其中 $t$ 表示操作类型,$t=1$ 时表示将编号为 $x$ 的结点的权值加上 $y$,$t=2$ 时表示询问编号为 $x$ 的结点是否为叶子节点。 输出格式: 对于每个操作 $t=2$,输出一个结果,表示询问的结点是否为叶子节点。 范围: $1≤n,m≤10^5$, $1≤w_i,y≤10^9$ 样例: 输入: 5 5 1 2 3 4 5 1 2 1 3 3 4 3 5 2 3 0 1 3 100 2 3 0 1 1 100 2 3 0 输出: 1 0 0 算法1: 暴力dfs,每次都重新遍历整棵树,时间复杂度 $O(nm)$ 时间复杂度: 最坏情况下,每次操作都要遍历整棵树,时间复杂度 $O(nm)$,无法通过此题。 算法2: 用一个 vector<int> sons[n+5] 来存储每个点的所有子节点,这样可以用 $O(n)$ 预处理出每个点的度 $deg_i$,如果 $deg_i=0$,则 $i$ 是叶子节点,否则不是。 对于每个操作,只需要更新叶子节点关系的变化就可以了。如果某个节点的度从 $1$ 变成 $0$,则该节点变成了叶子节点;如果某个节点的度从 $0$ 变成 $1$,则该节点不再是叶子节点。 时间复杂度: 每次操作的时间复杂度是 $O(1)$,总时间复杂度 $O(m)$,可以通过此题。 C++ 代码: (算法2)
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值