划分树——给出n个数,m个询问,每次询问区间[x,y]中第k小的数是多少。
#include <iostream> #include <cstdio> #include <algorithm> using namespace std; #define N 100500 #define MID ((l+r)>>1) int a[N],s[N],t[20][N],num[20][N],n,m; /* 建树:对于区间[l,r],首先通过对原数组的排序找到这个区间的中位数a[mid], 小于a[mid]的数划入他的左子树[l,mid-1],大于它的划入右子树[mid,r]。同时, 对于第i个数,记录在[l,i]区间内有多少数被划入左子树。最后, 对它的左子树区间[l,mid-1]和右子树区间[mid,r]递归的继续建树就可以了。 */ // void build(int lft,int rht,int ind) // { // if(lft==rht) return; // // int mid=MID(lft,rht); // int same=mid-lft+1,ln=lft,rn=mid+1; // for(int i=lft;i<=rht;i++) // if(valu[ind][i]<order[mid]) same--; // for(int i=lft;i<=rht;i++) // { // int flag=0; // if((valu[ind][i]<order[mid])||(valu[ind][i]==order[mid]&&same))//划分到左区间 // { // flag=1; // valu[ind+1][ln++]=valu[ind][i]; // lsum[ind][i]=lsum[ind][i-1]+valu[ind][i]; // if(valu[ind][i]==order[mid]) same--; // } // else //划分到右区间 // { // valu[ind+1][rn++]=valu[ind][i]; // lsum[ind][i]=lsum[ind][i-1]; // } // num[ind][i]=num[ind][i-1]+flag;//划分到左区间的个数 // } // build(lft,mid,ind+1); // build(mid+1,rht,ind+1); // } // int query(int st,int ed,int k,int lft,int rht,int ind)//子区间[st,ed],大区间[lft,rht],层数ind // { // if(lft==rht) return valu[ind][lft]; // // int mid=MID(lft,rht); // int lx=num[ind][st-1]-num[ind][lft-1];//在左区间不属于子区间的个数 // int ly=num[ind][ed]-num[ind][st-1];//有多少个进入左区间 // int rx=st-1-lft+1-lx;//在右区间不属于子区间的个数 // int ry=ed-st+1-ly;//有多少个进入右区间 // if(ly>=k) return query(lft+lx,lft+lx+ly-1,k,lft,mid,ind+1); // else // { // isum+=lsum[ind][ed]-lsum[ind][st-1]; // st=mid+1+rx; // ed=mid+1+rx+ry-1; // return query(st,ed,k-ly,mid+1,rht,ind+1); // } // } void Build(int c,int l,int r) { int lm=MID-l+1,lp=l,rp=MID+1; for(int i=l;i<=MID;i++) lm-=s[i]<s[MID]; for(int i=l;i<=r;i++) { if( i==l ) num[c][i]=0; else num[c][i]=num[c][i-1]; if( t[c][i]==s[MID] ) { if( lm ) { lm--; num[c][i]++; t[c+1][lp++]=t[c][i]; } else t[c+1][rp++]=t[c][i]; } else if( t[c][i]<s[MID] ) { num[c][i]++; t[c+1][lp++]=t[c][i]; } else t[c+1][rp++]=t[c][i]; } if( l<r ) Build(c+1,l,MID),Build(c+1,MID+1,r); } int Query(int c,int l,int r,int ql,int qr,int k) { if( l==r ) return t[c][l]; int s,ss; if( l==ql ) s=0,ss=num[c][qr]; else s=num[c][ql-1],ss=num[c][qr]-num[c][ql-1]; if( k<=ss ) return Query(c+1,l,MID,l+s,l+s+ss-1,k); else return Query(c+1,MID+1,r,MID+1+ql-l-s,MID+1+qr-l-s-ss,k-ss); } int main() { scanf("%d%d",&n,&m); for(int i=1;i<=n;i++) { scanf("%d",&a[i]); s[i]=t[0][i]=a[i]; } sort(s+1,s+1+n); Build(0,1,n); while( m-- ) { int l,r,k; scanf("%d%d%d",&l,&r,&k); printf("%d\n",Query(0,1,n,l,r,k)); } return 0; }