2019 hdu 多校四 K-th Closest Distance (主席树)

链接



题意:

给出一个数组,每次询问:l,r,p,k,数组中l到r区间内跟p的差的绝对值排名第k的。强制在线,l,r,p,k都要异或上一次的答案。



思路:

数组中的数不超过1e6,p也不超过1e6,那么可以不用离散,直接把数值当成权值建主席树就可以了。
对于查询,给定了p,那么就二分区间的半径,查询[p-R,p+R]中的数字是否大于等于k个,若大于则缩小半径,否则扩大半径,最后的半径就是答案了。


参考代码:

#include
#include 
using namespace std;
const int N=100005;
const int M=1e6;

int num[N];
int root[N];
int lchild[N*30],rchild[N*30],tsize[N*30];
int tot;

void update(int last,int cur,int l,int r,int k)
{
    lchild[cur]=lchild[last];
    rchild[cur]=rchild[last];
    tsize[cur]=tsize[last]+1;
    if(l==r)return;
    int mid=(l+r)>>1;
    if(k<=mid)update(lchild[last],lchild[cur]=++tot,l,mid,k);
    else update(rchild[last],rchild[cur]=++tot,mid+1,r,k);
}

int query(int last,int cur,int L,int R,int l,int r)
{
    if(L>=l&&R<=r){
        return tsize[cur]-tsize[last];
    }
    int mid=(L+R)>>1;
    if(r<=mid)return query(lchild[last],lchild[cur],L,mid,l,r);
    else if(l>mid) return query(rchild[last],rchild[cur],mid+1,R,l,r);
    else return query(lchild[last],lchild[cur],L,mid,l,mid)+query(rchild[last],rchild[cur],mid+1,R,mid+1,r);
}

int pre;
template <typename _Tp> inline _Tp read(_Tp&x){
    char c11=getchar(),ob=0;x=0;
    while(c11^'-'&&!isdigit(c11))c11=getchar();if(c11=='-')c11=getchar(),ob=1;
    while(isdigit(c11))x=x*10+c11-'0',c11=getchar();if(ob)x=-x;return x;
}

int main()
{
    int t;
    read(t);
    for(int ca=1;ca<=t;ca++){
        pre=tot=0;
        int n,m;
        read(n),read(m);
        for(int i=1;i<=n;i++){
            scanf("%d",&num[i]);
            update(root[i-1],root[i]=++tot,1,M,num[i]);
        }
        int L,R,p,d;
        for(int i=1;i<=m;i++){
            read(L),read(R),read(p),read(d);
            L^=pre;
            R^=pre;
            p^=pre;
            d^=pre;
            int l=0,r=M,ans=-1,mid;
            while (l<=r){
                mid=(l+r)>>1;
                if(query(root[L-1],root[R],1,M,max(1, p - mid), min(M, p + mid)) >= d){
                    ans=mid;
                    r=mid-1;
                } else{
                    l=mid+1;
                }
            }
            pre=ans;
            printf("%d\n",ans);
        }
    }
    return 0;
}

你可能感兴趣的:(数据结构)