今日头条2018校招后端方向（第二批）第一题 二分查找

原题链接 https://www.nowcoder.com/test/8537209/summary

题意 n个数  q个 查询  L,R,K  L到R区间内为K的数有多少个 数据范围  n <=300000,q<=300000 

解析  对于每次查询必须要 O(logn) 复杂度才行  所以想到二分查找  因为数据是不变的没有增删  所以我们对K进行排序 K值相同的对下标排序

然后 标记一下每个K值的出现范围(新的下标）s[K]到e[K]   然后二分查找 出区间内>=L的最左边位置 和区间内<=R最右边的位置  两个相减就是答案

AC代码

  

#include<bits/stdc++.h>
using namespace std;
const int maxn=3e6,inf=0x3f3f3f3f;
typedef long long ll;
struct node
{
    int x,val;
}a[maxn];
map<ll,int> s,e;
int cmp(node a,node b)
{
    if(a.val==b.val)
        return a.x<b.x;
    return a.val<b.val;
}
int main()
{
    int n,m,t;
    while(scanf("%d",&n)!=EOF)
    {
        for(int i=1;i<=n;i++)
        {
            scanf("%d",&a[i].val);
            a[i].x=i;
        }
        sort(a+1,a+n+1,cmp);
        for(int i=1;i<=n;i++)
            e[a[i].val]=i;
        for(int i=n;i>=1;i--)
            s[a[i].val]=i;
        scanf("%d",&m);
        while(m--)
        {
            int L,R,K;
            scanf("%d%d%d",&L,&R,&K);
            if(s[K]==0)
            {
                printf("0
");
                continue;
            }
            int l,r;
            l=s[K],r=e[K];
            int ansl=inf;
            while(l<=r)
            {
                int mid=(l+r)/2;
                if(a[mid].x>=L)
                {
                    ansl=min(mid,ansl);
                    r=mid-1;
                }
                else
                    l=mid+1;
            }
            l=s[K],r=e[K];
            int ansr=-1;
            while(l<=r)
            {
                int mid=(l+r)/2;
                if(a[mid].x<=R)
                {
                    ansr=max(ansr,mid);
                    l=mid+1;
                }
                else
                    r=mid-1;
            }
           if(ansl==inf||ansr==-1)
                printf("0
");
            else
                printf("%d
",ansr-ansl+1);
        }
    }
}