【大杀器】利用划分树秒杀区间内第k大的数

//本文首先发表于博客园,点击这里查看我的博客。  
最近看了一道题,大概就是给出一个序列,不断询问其子区间内第k大的数,下面是个截图

image

绕了一圈没找到中文版题目,if(你是大佬) then 去看截图;else{我来解释:给出一个整数n,和一个整数m,分别表示序列元素个数和询问次数,然后输入n个数和m个询问,每个询问包含3个数,分别是区间起止点(l和r)和k,求出区间内第k大的数并输出;}这是一道很简单的模板题,怎么解决呢?小编最初想到的是打暴力,正所谓暴力出奇迹,说不定可以成功,反正不会优化,先试试吧,直接把规定区间[l,r]排一次序了,然后在查找一遍第k大的数,但是毫无疑问,绝对会超时,怎样能减少时间复杂度呢?这是就请出了二分。

二分运用了分治的思想,不断将子区间分成两半,直到找到第k大的数,虽然很高效,但是面对这道题的多次询问,二分也只能表示手软,仍然过不了这道题。但是如果换成了线段树的结构,效率则会快很多,线段树看起来也用了分治的思想,把原来的整个序列都大约相等的长度分到左子树和右子树,不断分下去,直到全部分成叶子节点,在逐次确定下一层第k大的数在左子树还是右子树,虽然这种树能成功AC,但是并不是最优的,做题不能只讲求AC,下面就来讲一讲线段树的升级版——划分树。

什么是划分树?

划分树是一种基于线段树的数据结构,也利用了分治的思想,却比线段树高效很多,这是为什么?因为划分树又多了一个性质:在划分时不是随意划分,也不是排序后直接划分(因为这样会破坏原有结构),而是排序后仍保持原来的相对顺序再分到左右子树。

具体实现方法:

整个过程分为建树和查询两个阶段:

image

//copy来的图

1)建树:首先定义一个数组tree[30][1000]第一个维度表示层数,第二个维度表示这一层第i个数的值,用来表示这棵划分树,然后定义sorted[1000]数组,用来存储排序好的原序列,然后记录每一层前i个数有多少进入了下一层的左子树,存在toleft[30][1000]数组中,在建树中没用,但记录下来对查找时有用,用分治的思想分配左子树和右子树,将不大于中间值mid的数分配到左子树中去,大于中间值的分配到右子树中去,但有时为了左右尽可能个数相等,要把等于中间值的数两边都分配,于是定义same来存储多少等于中间值的数进入左子树,分配完毕后再递归分配左子树和右子树。身为递归,怎么也要有个出口吧,递归到叶子节点时就返回,即if(l==r) return;

2)查询:按照之前存储下的进入下一层左子树个数的数组toleft,可以计算出区间内第k大的数在当前节点的左子树还是右子树,并计算出下一层相应子树的左右边界,然后递归相应子树,同上,递归出口也是到达叶子节点时返回。详见注释……

废话不多说,代码呈上:

<pre style="color: rgb(0, 0, 0); font-family: "Courier New"; font-size: 12px; margin: 5px 8px; padding: 5px;">#include<iostream> #include<algorithm>
using namespace std; int tree[30][1000],sorted[1000],toleft[30][1000],n,m,ans;//tree和toleft的两个维度分别存储深度和序列,sorted存储的是排序好的序列
void buildtree(int l,int r,int dep)//构建划分树
{ if(l==r) return;//遇见叶子节点就返回
int mid=(l+r)/2;//二分
int same=mid-l+1;//same最终保存的是和中间值相同元素的个数,以便确定分到哪一区间
for(int i=l;i<=r;i++) if(tree[dep][i]<sorted[mid]) same--; int lpos=l;int rpos=mid+1;//左指针和右指针,并非常用的指针,是用来保存现在各区间内元素个数
for(int i=l;i<=r;i++)
{ if(tree[dep][i]<sorted[mid])//小于中间值
tree[dep+1][lpos++]=tree[dep][i];//分配到左子区间
else if(tree[dep][i]==sorted[mid]&&same>0)//等于中间值且相同个数大于0
{
same--;
tree[dep+1][lpos++]=tree[dep][i];//分配到右子区间
} else tree[dep+1][rpos++]=tree[dep][i];//剩下的分配到右子区间
toleft[dep][i]=toleft[dep][l-1]+lpos-l;//toleft数组记录这一层前i个数有多少个进入下一层的左子区间,查询时有用
}
buildtree(l,mid,dep+1);//构建左子区间(左子树)
buildtree(mid+1,r,dep+1);//构建右子区间(右子树)
} int search(int L,int R,int l,int r,int dep,int k)//查询第k大的数
{ if(l==r) return tree[dep][l];//查询到符合要求的叶子节点,返回相应的值
int mid=(L+R)/2;//L,R为大区间(主要是每个左子树,右子树的边界)
int cnt=toleft[dep][r]-toleft[dep][l-1];//求出[l,r]区间内有多少数进入下一层左子区间
if(cnt>=k)//第k大的数对应节点在左子树
{ int newl=L+toleft[dep][l-1]-toleft[dep][L-1];//求出下一层第k大的数所在区间边界
int newr=newl+cnt-1; return search(L,mid,newl,newr,dep+1,k);
} else//在右子树
{ int newr=r+toleft[dep][R]-toleft[dep][r];//求出下一层第k大的数所在区间边界
int newl=newr-(r-l-cnt); return search(mid+1,R,newl,newr,dep+1,k-cnt);
}
} int main()
{
cin>>n>>m; for(int i=1;i<=n;i++)
{
cin>>tree[0][I];
sorted[i]=tree[0][I];
}
sort(sorted+1,sorted+n+1);//为sorted数组排序
buildtree(1,n,0);//建树
int a,b,c; for(int i=1;i<=m;i++)
{
cin>>a>>b>>c;//输入询问
cout<<search(1,n,a,b,0,c)<<endl;//查询第k大的数
} return 0;
}</pre>

//额~好像忘了改scanf和printf了,没过别怪我……

©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 218,036评论 6 506
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 93,046评论 3 395
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 164,411评论 0 354
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 58,622评论 1 293
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 67,661评论 6 392
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 51,521评论 1 304
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 40,288评论 3 418
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 39,200评论 0 276
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 45,644评论 1 314
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 37,837评论 3 336
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 39,953评论 1 348
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 35,673评论 5 346
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 41,281评论 3 329
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 31,889评论 0 22
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 33,011评论 1 269
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 48,119评论 3 370
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 44,901评论 2 355

推荐阅读更多精彩内容