hdu6621--主席树

孤街醉人 提交于 2020-01-31 19:14:13

K-th Closest Distance

题意:给定长度为\(n\)的数列,现有m次查询,每组询问给\(l,r,p,k\),问对\(l<=i<=r\),\(|p-a[i]|\)的第k小值。数据强制在线。

题解:二分答案ans,用主席树查询\(a[l]\)\(a[r]\)之间\([p-ans,p+ans]\)的个数,个数为k即为答案。

​ 由于强制在线,所以建主席树不能离散化。可以动态空间直接建树。

#include<bits/stdc++.h>
using namespace std;
const int maxn=100005;
int n,m;
const int N=1e6;
struct node{
    int ls,rs,sum;
};
node tree[maxn*22];
int rt[maxn],tot=0;
int update(int l,int r,int x,int pre){
    int now=++tot;
    tree[now]=tree[pre];
    if(l==r){
        tree[now].sum++; return now;
    }
    int mid=(l+r)>>1;
    if(x>mid) tree[now].rs=update(mid+1,r,x,tree[pre].rs);
    else tree[now].ls=update(l,mid,x,tree[pre].ls);
    tree[now].sum++;
    return now;
}
int ask(int l,int r,int il,int ir,int root1,int root2){
    if(r<il||l>ir) return 0;
    if(l>=il&&r<=ir) return tree[root2].sum-tree[root1].sum;
    int mid=(l+r)>>1;
    return ask(l,mid,il,ir,tree[root1].ls,tree[root2].ls)+ask(mid+1,r,il,ir,tree[root1].rs,tree[root2].rs);
}
int jud(int len,int p,int rt1,int rt2){
    int l=max(1,p-len),r=min(N,p+len);
    return ask(1,N,l,r,rt1,rt2);
}
int solve(int l,int r,int p,int k){
    int rt1=rt[l-1],rt2=rt[r];
    int ll=0,rr=N;
    while(ll<rr){
        int mid=(ll+rr)>>1;
        if(jud(mid,p,rt1,rt2)>=k) rr=mid;
        else ll=mid+1;
    }
    return ll;
}
int main(){
    int T;
    cin >> T;
    while(T--){
        tot=0;
        tree[0].sum=tree[0].ls=tree[0].rs=0;
        scanf("%d%d",&n,&m);
        for(int i=1;i<=n;i++) {
            int x;
            scanf("%d",&x);
            rt[i]=update(1,N,x,rt[i-1]);
        }
        int las=0;
        while(m--){
            int l,r,p,k;
            scanf("%d%d%d%d",&l,&r,&p,&k);
            l^=las; r^=las; p^=las; k^=las;
            printf("%d\n",las=solve(l,r,p,k));
        }
    }
    return 0;
}
标签
易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!