HDU 6621 K-th Closest Distance(主席树+二分) 2019暑假杭电多校第四场

题目大意:

       T组,n个数字,m次查询,每次查询 [  L , R ] 这个区间距离 q 第K小的距离。

题目思路:

       考虑到K只有169范围,所以可以考虑主席树,主席树查询【L到R】区间中数值在【LL,RR】范围内的数字的个数。对于每组查询直接二分答案,判断p-mid,q+mid之间的数字够不够k个就行。

#include
#define ll long long
using namespace std;
const int MAXN=1e5+5;
int n,m;
int a[MAXN];
int L[MAXN*30],R[MAXN*30],T[MAXN],sum[MAXN*30];
int cnt=0;
int update(int pre,int l,int r,int x)   ///建主席树
{
    int rt=++cnt;
    L[rt]=L[pre];R[rt]=R[pre];sum[rt]=sum[pre]+1;
    int mid=(l+r)/2;
    if(lmid)ret+=query(R[u],R[v],mid+1,r,LL,RR);
    return ret;
}

int main()
{
    int t;
    scanf("%d",&t);
    while(t--)
    {
        int Max=-1;
        cnt=0;
        scanf("%d%d",&n,&m);
        for(int i=1;i<=n;i++){
            scanf("%d",&a[i]);
            Max=max(Max,a[i]);
        }
        int mm=Max;
        Max+=5;
        T[0]=0;
        for(int i=1;i<=n;i++){
            T[i]=update(T[i-1],1,Max,a[i]);
        }
        int now=0;
        while(m--)
        {
            int l,r,p,k;
            scanf("%d%d%d%d",&l,&r,&p,&k);
            l^=now,r^=now,p^=now,k^=now;
            int LL=0,RR=1e6+5,ans;
            while(LL<=RR)
            {
                int mid=(LL+RR)/2;
                if(query(T[l-1],T[r],1,Max,max(1,p-mid),min(mm,p+mid))>=k ){
                    RR=mid-1;
                    ans=mid;
                }
                else LL=mid+1;
            }
            printf("%d\n",ans);
            now=ans;
        }
    }
}

 

你可能感兴趣的:(tree)