玲珑杯Round20PE

#include <iostream>
#include <algorithm>
#include <cstring>
#include <cstdio>
#include <cstdlib>
#include <cmath>
#define maxn 400100

using namespace std;

struct Data
{
    int val,id;
};

struct Tree
{
    int l,r,sum;
};

Data a[maxn];
Tree t[maxn*100];

int Rank[maxn],Root[maxn],cntt;

bool cmp(Data a,Data b)
{
    return a.val<b.val;
}

void init()
{
    cntt=1;
    Root[0]=0;
    t[0].l=t[0].r=t[0].sum=0;
    return;
}

void update(int val,int&root,int l,int r)
{
    t[cntt]=t[root];
    t[cntt].sum++;
    root=cntt;
    cntt++;

    if (l==r) return;

    int mid=(l+r)/2;
    if (val<=mid) update(val,t[root].l,l,mid);
    else update(val,t[root].r,mid+1,r);
    return;
}

int query(int rl,int rr,int k,int l,int r)
{
    if (l==r) return l;

    int sum=t[t[rr].l].sum-t[t[rl].l].sum;
    int mid=(l+r)/2;

    if (k<=sum) return query(t[rl].l,t[rr].l,k,l,mid);
    else return query(t[rl].r,t[rr].r,k-sum,mid+1,r);
}

int findx(int rl,int rr,int x,int l,int r)
{
    if (l==r) return t[rr].sum-t[rl].sum;

    int sum=t[t[rr].l].sum-t[t[rl].l].sum;
    int mid=(l+r)/2;

    if (x<=mid) return findx(t[rl].l,t[rr].l,x,l,mid);
    else return sum+findx(t[rl].r,t[rr].r,x,mid+1,r);
}

int trans(int x,int n)
{
    a[n+1].val=0x3f3f3f3f;
    int l=1,r=n+1;
    while(l<r)
    {
        int mid=(l+r)/2;
        if (a[mid].val<=x) l=mid+1;
        else r=mid;
    }
    return l-1;
}

int main()
{
    int n,m;
    while(~scanf("%d %d",&n,&m))
    {
        for (int k=1;k<=n;k++)
            {scanf("%d",&a[k].val);a[k].id=k;}

        sort(a+1,a+n+1,cmp);

        for (int k=1;k<=n;k++)
            Rank[a[k].id]=k;

        init();

        for (int k=1;k<=n;k++)
        {
            Root[k]=Root[k-1];
            update(Rank[k],Root[k],1,n);
        }

        for (int k=1;k<=m;k++)
        {
            int l,r,x,k1,k2,kx,tranx;
            int ans1,ans2;
            scanf("%d %d %d %d %d",&l,&r,&x,&k1,&k2);

            int st,en;
            st=query(Root[l-1],Root[r],1,1,n);
            en=query(Root[l-1],Root[r],r-l+1,1,n);
            st=a[st].val;en=a[en].val;

            if (x<st)
            {
                ans1=-1;
                if (k2<=r-l+1) ans2=a[query(Root[l-1],Root[r],k2,1,n)].val;
                else ans2=-1;
            }
            else if (x>en)
            {
                ans2=-1;
                if (k1<=r-l+1) ans1=a[query(Root[l-1],Root[r],k1,1,n)].val;
                else ans1=-1;
            }
            else
            {
                tranx=trans(x,n);
                kx=findx(Root[l-1],Root[r],tranx,1,n);
                if (k1<=kx) ans1=a[query(Root[l-1],Root[r],k1,1,n)].val;
                else ans1=-1;

                if (k2+kx<=r-l+1) ans2=a[query(Root[l-1],Root[r],k2+kx,1,n)].val;
                else ans2=-1;
            }

            printf("%d %d\n",ans1,ans2);
        }
    }
    return 0;
}

这题目的关键就是如果知道x在l，r中，我们要求<=x是第几个

这里加了两个函数，第一个transx是找第一个>=x的值离散化的值-1

findx就是找x(离散化过的)的位置