划分树——求区间第k大值
2012-10-06 18:34
302 查看
题目:K-th Number
虽说是第一次就AC,但到网上去搜罗更优代码时,却幸运的发现,我的AC代码是错误的,只不过数据太水。我考虑掉一种情况,当有多个数和中位数相同时,如果从左到右一次将小于等于中位数的数划入左子树,可能导致区间右端的哪些比中位数小的数被划入右子树。例如,数据为1 2 2 2 1时,第二个2应划入右子树,最后一个1应划入左子树。
虽说是第一次就AC,但到网上去搜罗更优代码时,却幸运的发现,我的AC代码是错误的,只不过数据太水。我考虑掉一种情况,当有多个数和中位数相同时,如果从左到右一次将小于等于中位数的数划入左子树,可能导致区间右端的哪些比中位数小的数被划入右子树。例如,数据为1 2 2 2 1时,第二个2应划入右子树,最后一个1应划入左子树。
#include <iostream> #include <cstdio> #include <cstring> #include <algorithm> using namespace std; #define MAXN 100010 #define lson l,m,d+1 #define rson m+1,r,d+1 int a[20][MAXN],sorted[MAXN],toLeft[20][MAXN]; void build(int l,int r,int d) { if(l==r) return ; int m=(l+r)>>1,p1=l-1,p2=m; int mv=sorted[m]; if(l==1) toLeft[d][0]=0; int lsame=m-l+1; for(int i=l;i<=r;++i) if(a[d][i]<mv) --lsame;//找出左子树可容纳多少中位数 for(int i=l;i<=r;++i) { toLeft[d][i]=toLeft[d][i-1]; if(a[d][i]<mv) ++toLeft[d][i],a[d+1][++p1]=a[d][i]; else if(a[d][i]>mv) a[d+1][++p2]=a[d][i]; else { if(lsame) ++toLeft[d][i],a[d+1][++p1]=a[d][i],--lsame;//左子树还可容纳中位数 else a[d+1][++p2]=a[d][i]; } } build(lson); build(rson); } int query(int k,int x,int y,int l,int r,int d) { if(l==r) return sorted[l]; int l1=toLeft[d][x-1]-toLeft[d][l-1],l2=toLeft[d][y]-toLeft[d][x-1];//l1=[l,x-1]上划入左子树的数目,l2=[x,y]上划入左子树的数目 int m=(l+r)>>1; int r1=x-l-l1,r2=y-x+1-l2;//r1=[l,x-1]上划入右子树的数目,r2=[x,y]上划入右子树的数目 if(k<=l2) return query(k,l+l1,l+l1+l2-1,lson); else return query(k-l2,m+1+r1,m+r1+r2,rson); } int main() { ios::sync_with_stdio(false); int n,m; cin >> n >> m; for(int i=1;i<=n;++i) cin >> a[1][i],sorted[i]=a[1][i]; sort(sorted+1,sorted+1+n); build(1,n,1); while(m--) { int a,b,k; cin >> a >> b >> k; cout << query(k,a,b,1,n,1) << endl; } return 0; }
相关文章推荐
- 划分树学习小记 Poj 2104+Poj 2761+Hdu 2665 (区间第k大数)
- 划分树的用法(一):查询区间第K大值值(poj2104)
- 划分树的学习(求区间第k大的数字)
- [NBUT 1458 Teemo]区间第k大问题,划分树
- 划分树的用法(一):查询区间第K大值值(poj2104)
- POJ 2104 区间第K大值(划分树做法)
- hdu 4417 Super Mario (二分法 + 划分树求区间第K大)
- 【专题】划分树-->求区间第K大的数
- POJ 2104 区间第K大值(划分树做法)
- 查询区间第k大 POJ2104 暴力 or 划分树 or 归并树
- 划分树求区间第k值
- 划分树(基本用法是求给定区间的第k大的值)
- HDU 4251 The Famous ICPC Team Again 划分树 区间第K大
- 划分树--区间第K元素模板
- poj2104 求区间第k大数(划分&&主席--待补)
- POJ2104:划分树求区间第k小
- hdu3437 划分树 区间内小于第K大的值得和
- HDU 3473 Minimum Sum (划分树求区间第k大带求和)(转)
- 划分树的用法(一):查询区间第K大值值(poj2104)
- hdu 3727 Jewel 划分数+树状数组 求区间和当前段的第k大数