主席树模板

 1 #include<bits/stdc+.h>
 2 const int maxn=200005;
 3 int sz,T,n,m,rt[maxn],tot,lc[maxn*20],rc[maxn*20],sum[maxn*20],a[maxn],b[maxn];
 4 using namespace std;
 5 void update(int &o,int l,int r,int last,int v){
 6     //o是当前结点 last是上一次更改过的结点.
 7     //将每一个结点插到该插好的地方.
 8     o=++tot;
 9     lc[o]=lc[last];
10     rc[o]=rc[last];
11     //此处借用上一次的结点
12     sum[o]=sum[last]+1;
13     //因为多插入了一个结点 所以前缀和数组需要加上一个1
14     //同时sum 数组起到一个前缀和的作用
15     if(l==r) return;
16     int mid=(l+r)>>1;
17     if(v<=mid) update(lc[o],l,mid,lc[last],v);
18     else update(rc[o],mid+1,r,rc[last],v);
19     //把每一个数安装到它应该在的地方.
20 }
21 int query(int ql,int qr,int l,int r,int k){
22     if(l==r) return l;    
23     int mid=(l+r)>>1;
24     //和splay 类似的查询 
25     int cnt=sum[lc[qr]]-sum[lc[ql]];
26     if(cnt>=k) return query(lc[ql],lc[qr],l,mid,k);
27     else return query(rc[ql],rc[qr],mid+1,r,k-cnt);
28 }
29 int main()
30 {
31      scanf("%d%d",&n,&m);
32      for(int i=1;i<=n;i++) { scanf("%d",&a[i]);b[i]=a[i];}
33      sort(b+1,b+n+1);
34      sz=unique(b+1,b+n+1)-(b+1);
35      tot=0;
36      rt[0]=++tot;
37      //每一棵子树的根结点
38      //build(rt[0],1,sz);
39      for(int i=1;i<=n;i++) a[i]=lower_bound(b+1,b+sz+1,a[i])-b;  //离散化 
40      for(int i=1;i<=n;i++) update(rt[i],1,sz,rt[i-1],a[i]);      
41      //建造每一棵线段树   
42      for(int i=1;i<=m;i++){
43        int l,r,k;scanf("%d%d%d",&l,&r,&k);
44        printf("%d\n",b[query(rt[l-1],rt[r],1,sz,k)]);
45      } 
46    return 0;
47 }