【模板】区间第k小

我实在是太弱了现在才会这个东西QAQ。

主席树做法。

一张关于主席树的无字说明

线段树\(2\)是只单点修改了实心酒红色点的线段树\(2\)，线段树\(2\)中的蓝色节点实际上就是线段树\(1\)的蓝色节点，我们只是把地址复制过来了。

我们多开了一个线段树，但是节点数量却只多了\(log\)层，那么对于\(n\)的历史版本保留就提供了\(O(nlogn)\)的数据结构啦。

具体代码怎么实现？我研究了一下，发现实际写出来的代码和大家常写的代码是一样的。实现其实非常简单，只需要在普通线段树上稍微修改一下。

如何访问历史版本？我们可以开一个\(rt(i)\)数组表示\(i\)号版本的根节点的编号，对于线段树每个节点我们要记录\(seg_L,seg_R,data\)，递归时只要访问\(seg_{L/R}\)的值就好了，不需要很大的修改。

其实我还想问一个问题，就是我发现这种主席树支持单点修改，区间查询，但是我想问如果是区间修改，而且是\(n\times n\)的修改怎么办？因为我发现修改一个点就一定会产生新的\(n\)个节点了。

实际上主席树就是让整个线段树有了两维下标，所以对于什么二维数点啊什么什么之类的也有用。

这个问题以后再解决，那么有了主席树工具之后怎么解此题？

不是两维下标吗？建立一个下标是值域和位置的线段树(按照给的序列的顺序建立主席树，主席树存储的东西是值域内有多少个数)，然后区间的限制可以直接是\(rt_R-rt_{L-1}\)很方便地查询。

实际上，主席树教会我们如何弄一个可持久化数组！有了可持久化数组，什么可持久化数据结构不行！~~平衡树呢:)~~

#include<bits/stdc++.h>

using namespace std;typedef long long ll;
#define RP(t,a,b)  for(register int t=(a),edd=(b);t<=edd;++t)
#define midd register int mid=(l+r)>>1
#define TMP template < class ccf >
#define lef l,mid,seg[pos].l
#define rgt mid+1,r,seg[pos].r
TMP inline ccf qr(ccf b){
    register char c=getchar();register int q=1;register ccf x=0;
    while(c<48||c>57)q=c==45?-1:q,c=getchar();
    while(c>=48&&c<=57)x=x*10+c-48,c=getchar();
    return q==-1?-x:x;}
TMP inline ccf Max(ccf a,ccf b){return a<b?b:a;}
TMP inline ccf Min(ccf a,ccf b){return a<b?a:b;}
TMP inline ccf Max(ccf a,ccf b,ccf c){return Max(a,Max(b,c));}
TMP inline ccf Min(ccf a,ccf b,ccf c){return Min(a,Min(b,c));}
TMP inline void READ(ccf* _arr,ccf* _b,int _n){RP(t,1,_n) _b[t]=_arr[t]=qr((ccf)1);}
//----------------------template&IO---------------------------
#define pushup(pos) (seg[pos].data=seg[seg[pos].l].data+seg[seg[pos].r].data)
const int maxn=2e5+5;
struct SEG{int l,r,data;}seg[maxn*81];//log2e5=17  ,  17*4=75

int rt[maxn];int data[maxn],sav[maxn];int sz,cnt;
int n,m;

void build0(int l,int r,int pos){
    if(l==r)return;midd;
    seg[pos].l=++cnt;seg[pos].r=++cnt;
    build0(lef);build0(rgt);
}

void build(int last,int k,int l,int r,int pos){
    if(l==r)  { seg[pos].data=seg[last].data+1; return; }midd;
    if(k<=mid){seg[pos].l=++cnt;seg[pos].r=seg[last].r;build(seg[last].l,k,lef);}
    else      {seg[pos].l=seg[last].l;seg[pos].r=++cnt;build(seg[last].r,k,rgt);}
    pushup(pos);
}

int a(int L,int R,int l,int r,int k){
    if(l>=r) return l;
    register int t=seg[seg[R].l].data-seg[seg[L].l].data;midd;
    if(t>=k) return a(seg[L].l,seg[R].l,l,mid,k);
    else return a(seg[L].r,seg[R].r,mid+1,r,k-t);
}

int main(){
#ifndef ONLINE_JUDGE
    freopen("in.in","r",stdin);
    freopen("out.out","w",stdout);
#endif
    n=qr(1);m=qr(1);
    READ(data,sav,n);
    sort(sav+1,sav+n+1);
    sz=unique(sav+1,sav+n+1)-sav-1;
    RP(t,1,n) data[t]=lower_bound(sav+1,sav+sz+1,data[t])-sav;
    cnt=rt[0]=1;build0(1,sz,1);
    RP(t,1,n){rt[t]=++cnt;build(rt[t-1],data[t],1,sz,rt[t]);}
    RP(t,1,m){
    register int t1,t2,t3;
    t1=qr(1);t2=qr(1);t3=qr(1);
    printf("%d\n",sav[a(rt[t1-1],rt[t2],1,sz,t3)]);
    }
    return 0;
}

【模板】区间第k小

【模板】区间第k小

猜你喜欢