主席树模板

Posted 2021-03-08 wsy107316

 

 

 

 AC_Code

 1 #include <iostream>
 2 #include <cstdio>
 3 #include <vector>
 4 #include <cstring>
 5 #include <algorithm>
 6 using namespace std;
 7 typedef long long ll;
 8 #define rep(i,first,last) for(int i=first;i<=last;i++)
 9 #define dep(i,first,last) for(int i=first;i>=last;i--)
10 const int maxn=1e5+6;
11 int n,m,cnt,root[maxn],a[maxn],x,y,k;//root为根节点编号
12 struct node{int l,r,sum;}T[maxn*50];//l:左儿子编号,r:右儿子编号,sum:值域元素个数和,开40~60倍
13 vector<int>v;
14 int getid(int x){ return lower_bound(v.begin(),v.end(),x)-v.begin()+1;}
15 void updata(int l,int r,int &x,int y,int pos){
16     T[++cnt]=T[y],T[cnt].sum++,x=cnt;
17     if(l==r) return ;
18     int mid=(l+r)>>1;
19     if( pos>mid ) updata(mid+1,r,T[x].r,T[y].r,pos);
20     else updata(l,mid,T[x].l,T[y].l,pos);
21 }
22 int query(int l,int r,int x,int y,int k){//传的x,y为编号
23     if( l==r )return l;
24     int mid=(l+r)>>1;
25     int sum=T[T[y].l].sum-T[T[x].l].sum;
26     if( sum>=k ) return query(l,mid,T[x].l,T[y].l,k);
27     else return query(mid+1,r,T[x].r,T[y].r,k-sum);
28 }
29 
30 int main()
31 {
32     scanf("%d%d",&n,&m);
33     rep(i,1,n) scanf("%d",&a[i]),v.push_back(a[i]);
34     sort(v.begin(),v.end()),v.erase(unique(v.begin(),v.end()),v.end());
35     rep(i,1,n) updata(1,n,root[i],root[i-1],getid(a[i]));
36     rep(i,1,m){
37         scanf("%d%d%d",&x,&y,&k);
38         printf("%d
",v[query(1,n,root[x-1],root[y],k)-1]);
39     }
40     return 0;
41 }