一道裸的划分树,求区间第k大数
第一次写,没有考虑清楚,错了无数次
错误原因:没有考虑到有很多等于中位数的数,然后在末尾有小于中位数的数,
被fxz发现了,不然还不知道要查多久呢。。最后是加了一个sm计数,记录与中位数相等的数有几个可以到左边的
划分树的思想:
也是一颗线段树,选取一个中位数,把比他小的放在左边,比他大的放在右边,原理跟快排相似,
左右两边的数的相对位置要保持不变,这样查询的时候就可以在子树上定位区间了。与中位数相等的数要特殊处理下------
有两个数组,val[i][j]表示第i层第j个元素的值,onleft[i][j]表示第i层在,在这个子树的区间[L,R]内,[L,j]中多少个数放到了左边(左子树)
建好树之后查询:
对于区间[l,r],查询第K大数,如果[l,r]区间到左边的数个数大于等于K那么应该到左子树去查找,
反之到右子树查找
到子树上查找的时候要重定位 比如 l,r区间有n个树到了左边 n<k
那么 应该到右子树查k-n大的数, l新的位置是[L,l-1]区间到右边的数的个数+mid+1,r的新位置是[L,r]区间到右边的数的个数+mid (不加1是因为mid+1是第一个位置)
反之类似
#include<algorithm> #include<cstdio> #include<cstring> #define maxn 100007 using namespace std; int ord[maxn],onleft[18][maxn],val[18][maxn]; void build(int fl,int l,int r){ if(l == r) return; int mid = (l+r)/2,fn=fl+1; int le=l,ri=mid+1,sm=mid+1-l; for(int i = l;i<=r;i++)//记录左子树可以放几个等于中位数的数 if(val[fl][i]<ord[mid])sm--; for(int i=l;i<=r;i++){ if(i==l)onleft[fl][i]=0; else onleft[fl][i]=onleft[fl][i-1]; if(val[fl][i]<ord[mid]||(val[fl][i]==ord[mid]&&sm>0)){ if(ord[mid]==val[fl][i])sm--; onleft[fl][i]++; val[fn][le++]=val[fl][i]; } else val[fn][ri++]=val[fl][i]; } build(fl+1,l,mid),build(fl+1,mid+1,r); } int query(int fl,int L,int R,int l,int r,int k){ if(l == r) return val[fl][l]; int onr = onleft[fl][r],onl=0; if(l>L) onl=onleft[fl][l-1]; int n = onr-onl,mid=(L+R)/2; if(n >= k)return query(fl+1,L,mid,L+onl,L+onr-1,k);//左子树 else return query(fl+1,mid+1,R,mid+1+(l-L-onl),mid+1+(r-L-onr),k-n);//右子树 } int main(){ int n, t,q; scanf("%d",&t); memset(onleft,0,sizeof(onleft)); while(t--){ scanf("%d%d",&n,&q); for(int i = 1;i <= n; i++){ scanf("%d",&val[0][i]); ord[i] = val[0][i]; } sort(ord+1,ord+n+1); build(0,1,n); int l,r,k; for(int i = 0;i < q; i++){ scanf("%d%d%d",&l,&r,&k); printf("%d\n",query(0,1,n,l,r,k)); } } return 0; }