RMQ问题详解(线段树,树状数组,ST,RMQ转LCA,Splay(伸展树))

由于当年的百度空间和网易博客上发布的内容都因为这两个博客的停止维护都不在啦,现在上了大学,就读的也是计算机专业,有些舍不得以前在这两个博客上发的文章,就只好手动搬家过来简书这边啦~ 希望能够帮助到正在学习信息学竞赛的同学们哦~ 哈哈哈,有些内容毕竟是高中时代写的,还有些稚嫩,还请大家多多包涵哦。

RMQ问题,即Range Maximum/Minimum Query(区间最值查询问题),指对于一个有序序列,回答若干区间的数值最值的查询。

注:下面贴出代码均只支持查询。

主要算法(或数据结构):
1.线段树:
一种二叉搜索树,其每个节点均为一个区间,支持大多数快速区间操作,查询,如区间[1,10]建树如下:

f603918fa0ec08fa2a2c927158ee3d6d55fbda3c.jpg

时间复杂度: 预处理O(n), 查询O(log n) 总复杂度O(n+q log n)
空间复杂度:O(n) (常数较大)
优点:容易理解,编写,适用范围广
缺点:在支持区间修改操作时代码较长。
代码(指针,较慢):

#include <cstdio>
#include <algorithm>



using namespace std;



#define MAXN 100001



struct node {
 node *left,*right;
 int l,r,max;
};



node *roof;



int n,m;
int a[MAXN];



int Build(int l,int r,node *p){
 (*p).l=l;
 (*p).r=r;
 if (l==r){
  (*p).left=(*p).right=NULL;
  (*p).max=a[l];
  return 0;
 } 
 Build(l,(l+r)>>1,(*p).left=new(node));
 Build(((l+r)>>1)+1,r,(*p).right=new(node));
 (*p).max=max((*(*p).left).max,(*(*p).right).max);
 return 0;
}



int get_max(int l,int r,node *p){
 if (l==(*p).l&&r==(*p).r){
  return (*p).max;
 }
 int mid=((*p).l+(*p).r)>>1;
 if (r<=mid){
  return get_max(l,r,(*p).left);
 }
 if (l>mid){
  return get_max(l,r,(*p).right);
 }
 return max(get_max(l,mid,(*p).left),get_max(mid+1,r,(*p).right));
}



int main(){
 scanf("%d%d",&n,&m);
 for (int i=0;i++<n;){
  scanf("%d",&a[i]);
 }
 Build(1,n,roof=new(node));
 while (m--){
  int l,r;
  scanf("%d%d",&l,&r);
  printf("%d\n",get_max(l,r,roof));
 }
 return 0;
} 

之前去膜拜了ZKW大神的线段树,结果看了半天都没弄懂自底向上的线段树,所以自己写了一个从中间查询的代码,用一个辅助数组first[i]来储存以i为左端点的区间节点,从中间开始查找覆盖区间,速度还可以(就是遇到区间修改就悲剧了)

代码2:

#include <cstdio>

#include <cstring>

#include <algorithm>

#include <cmath>



using namespace std;



#define MAXN 100001



struct node {

int l,r,max;

} t[MAXN*5];



int a[MAXN];

int n,m;

int first[MAXN];



void Build(int l,int r,int rank){

if (!first[l]){

first[l]=rank;

}

t[rank].l=l;

t[rank].r=r;

if (l==r){

t[rank].max=a[l];

return ;

}

int mid=(l+r)>>1,left=rank<<1,right=left^1;

Build(l,mid,left);

Build(mid+1,r,right);

t[rank].max=max(t[left].max,t[right].max);

}



int getmax(int l,int r){

int rec=0,i=l;

while (i<=r){

int j=first[i];

while (t[j].r>r) j=j<<1;

rec=max(rec,t[j].max);

i=t[j].r+1;

}

return rec;

}



int main(){

memset(first,0,sizeof(first));

freopen("input.txt","r",stdin);

freopen("output.txt","w",stdout);

scanf("%d%d",&n,&m);

for (int i=0;i++<n;){

scanf("%d\n",&a[i]);

}

Build(1,n,1);

while (m--){

int l,r;

scanf("%d%d",&l,&r);

printf("%d\n",getmax(l,r));

}

fclose(stdin);

fclose(stdout);

return 0;

}

花了好几天终于把ZKW线段树膜拜完了~~,速度超快,常数真的很小。。。不废话了,贴代码:

#include <cstdio>

#include <algorithm>



using namespace std;



#define MAXN 200000

#define MAXD 20

#define D 1



struct node {

  int l,r,max;

} tree[MAXN*2];



int z[MAXD];

int N,n,m,M;

int a[MAXN];



void Build(int l,int r,int t){

    tree[t].l=l;

    tree[t].r=r;

    if (l==r){

      tree[t].max=a[l];

      if (l==1){

        M=t-1;

      }

      return ;

    }

  Build(l,(l+r)>>1,t<<1);

  Build(((l+r)>>1)+1,r,(t<<1)^1);

  tree[t].max=max(tree[t<<1].max,tree[(t<<1)^1].max);

}



int getmax(int l,int r){

  int rec=0;

  l--,r++;

  l+=M,r+=M;

  while ((l^r)!=1){

    if (!(l&1)) rec=max(rec,tree[l^1].max);

    if (r&1) rec=max(rec,tree[r^1].max);

    l>>=1;

    r>>=1;

  }

  return rec;

}



int main(){

  freopen("input.txt","r",stdin);

  freopen("output.txt","w",stdout);

  z[0]=1;

  for (int i=0;i++<MAXD-1;){

    z[i]=z[i-1]*2;

  }

  scanf("%d%d",&n,&m);

  for (int i=0;i++<n;){

    scanf("%d",&a[i+D]);

  }

  for (N=0;z[N]<=n+D;N++) ;

  Build(1,z[N],1);

  while (m--){

    int l,r;

    scanf("%d%d",&l,&r);

    printf("%d\n",getmax(l+D,r+D));

  }

  fclose(stdin);

  fclose(stdout);

  return 0;

}

2.树状数组(Binary Index Tree)。
树状数组是线段树一种小巧的替代品,也支持某一些区间查询修改操作(如:区间求和,RMQ等)
预备函数:lowbit (x) = ((~x)+1)&x
建立:若以T[]为索引数组,则t[i]表示区间[i-lowbit(i)+1,i] 。索引数组表示区间如下(如表示[1,10]):

6d81800a19d8bc3e4956871c838ba61ea9d3458e.jpg

查询:从i=r开始扫描,如果区间[i-lowbit(i)+1,i] 在查询区间内,则ans=max/min(ans,t[i]) i=i-lowbit(i) 否则 ans=max/min(ans,a[i]) i-- 知道整个区间扫面完成为止。
时间复杂度:预处理:O(n)或O(n log n) 查询:O(log^2 n) 总复杂度:O(n + q log^2 n)或O(n log n + q log^2 n)
空间负责度:O (n)(常数较小)
优点:节省空间,代码简短,常数小
缺点: 查询复杂度较高,在支持修改情况下复杂度较高,使用范围较小.
代码:

#include <cstdio>
#include <algorithm>



using namespace std;



#define MAXN 100001



int lowbit(int x){
 return ((~x)+1)&x;
}



int t[MAXN];
int a[MAXN];
int n,m;



int get_max(int l,int r){
 int rec=-0x7fffffff;
 if (!r){
  return rec;
 }
 int i=r;
 while (i>=l){
  int j=i-lowbit(i)+1;
  if (j>=l){
   rec=max(rec,t[i]);
   i=j-1;
  } else {
   rec=max(rec,a[i]);
   i--;
  }
 }
 return rec;
}



int main(){
 scanf("%d%d",&n,&m);
 for (int i=0;i++<n;){
  scanf("%d",&a[i]);
  int l=i-lowbit(i)+1,r=i-1;
  t[i]=max(a[i],get_max(l,r));
 }
 while (m--){
  int l,r;
  scanf("%d%d",&l,&r);
  printf("%d\n",get_max(l,r));
 }
 return 0;
}

3.ST算法:
该算法以倍增思想为基础,支持离线RMQ问题。
大意:
定义f[i][j]表示区间[i,i+2^j-1]
构建时使用动态规划思想:
f[i][j]=max/min(f[i][j-1] ,f[i+2^(j-1)][j-1])
对于某区间[l,r]支持O(1)RMQ查询
k=int(log10(r-l+1)/log10(2))
max (l,r)=max(f[l][k],f[r-EXP[k]+1][k])
空间复杂度:O(n log n)
时间复杂度:预处理O(n log n) 查询O(1) 总复杂度:O(n log n + q)
优点:代码简洁,快速
缺点:无法支持修改操作
代码:

#include <cstdio>
#include <algorithm>
#include <cmath>



using namespace std;



#define MAXN 100001



int f[MAXN][20];
int a[MAXN];
int EXP[20];



int m,n;



int main(){
 EXP[0]=1;
 for (int i=0;i++<19;){
  EXP[i]=EXP[i-1]*2;
 }
 scanf("%d%d",&n,&m);
 for (int i=0;i++<n;){
  scanf("%d",&a[i]);
  f[i][0]=a[i];
 }
 int i=1,j=2;
 while (j<=n){
  for (int h=0;h++<n;){
   if (h+j-1<=n){
    f[h][i]=max(f[h][i-1],f[h+(j/2)][i-1]);
   } else break;
  }
  i++;
  j*=2;
 }
 while (m--){
  int l,r;
  scanf("%d%d",&l,&r);
  int k=int(log10(r-l+1)/log10(2));
  printf("%d\n",max(f[l][k],f[r-EXP[k]+1][k]));
 }
 return 0;
}

4.将RMQ转LCA问题解决
建立Cartesian Tree,将区间问题转为树上的LCA(最近公共祖先问题解决)。LCA可用在线算法O(q log n)或Tarjan O(q)解决
空间复杂度:O(n)
时间复杂度:O(n+q)
优点:复杂度低
缺点:在有元素重复情况下较为麻烦,使用递归编写容易栈溢出

注:Cartesian Tree
如:对于某序列6 9 7 5 3 4 2建树如下:

8b13632762d0f703e9e6140a09fa513d2797c594.jpg

此时,某个区间[l,r]的最值,即l,r在树上对应元素的LCA的权.

5.splay(伸展树)维护区间求RMQ

用数列建立splay,关键字为某权值在序列中的位置,splay的某个节点上顺便储存以该节点为根的子树的权值最值,在进行splay(),ratote()的时候顺便维护,查询区间[l,r]时用splay()将关键字l-1的节点提到根,将关键字r+1的节点提到根的右子树,那么,查询结果就是根右子树的左子树上维护的信息。(注意,在序列中要另加两个节点0和n+1以便维护)

空间复杂度:O(n)

时间复杂度:O(q log n)(平摊)

优点:方便,支持区间插入,翻转,删除等线段树难以实现的操作。

缺点:时间常数较大。

代码:


#include <cstdio>

#include <algorithm>

#include <cstring>

 

using namespace std;

 

#define MAXN 100010

 

int a[MAXN];

 

struct node {

    int t,MAX;

    node *left,*right;

};

node *roof,*bank=new(node);

 

int n,m;

 

void zag(node* &t) {

    node*k=t->right;

    t->right=k->left;

    t->MAX=max(max(t->left->MAX,t->right->MAX),a[t->t]);

    k->left=t;

    k->MAX=max(max(k->left->MAX,k->right->MAX),a[k->t]);

    t=k;

}

 

void zig(node* &t) {

    node*k=t->left;

    t->left=k->right;

    t->MAX=max(max(t->left->MAX,t->right->MAX),a[t->t]);

    k->right=t;

    k->MAX=max(max(k->left->MAX,k->right->MAX),a[k->t]);

    t=k;

}

 

void Merge(node *u,node* &t,bool flag) {

    if (!t) {

       t=u;

        return ;

    }

    if (flag) {

       t->left=u;

       t->MAX=max(t->MAX,u->MAX);

       zig(t);

    }else {

       t->right=u;

       t->MAX=max(t->MAX,u->MAX);

       zag(t);

    }

}

 

void splay(int k,node* &t) {

    node*l=NULL,*r=NULL;

    while (t!=bank&&t->t!=k) {

        if (k<t->t) {

           if(t->left!=bank&&k<t->left->t) {

              zig(t);

           }

           node*u=t->left;

           t->MAX=max(t->right->MAX,a[t->t]);

           t->left=bank;

           Merge(t,r,true);

           t=u;

       } else {

           if(t->right!=bank&&k>t->right->t) {

              zag(t);

           }

           node*u=t->right;

           t->MAX=max(t->left->MAX,a[t->t]);

           t->right=bank;

           Merge(t,l,false);

           t=u;

       }

    }

    node*u;

    if (l) {

       u=t->left;

    t->left=l;

     l->right=u;

     l->MAX=max(l->MAX,u->MAX);

    }

    if (r) {

       u=t->right;

     t->right=r;

        r->left=u;

     r->MAX=max(r->MAX,u->MAX);

    }

    t->MAX=max(max(t->left->MAX,t->right->MAX),a[t->t]);

}

 

void INSERT(int k) {

    node*t=roof,*T=NULL;

    bool flag;

    while (t!=bank) {

       T=t;

        if (k<t->t) flag=false,t=t->left

       ;  else flag=true,t=t->right;

    }

    t=new(node);

    t->left=t->right=bank;

    t->MAX=a[k];

    t->t=k;

    if (T) {

        if (flag) T->right=t

       ;  else T->left=t;

    } else roof=t;

}

 

int main() {

    freopen("input.txt","r",stdin);

    freopen("output.txt","w",stdout);

    scanf("%d%d",&n,&m);

    for (int i=0;i++<n;) {

       scanf("%d",&a[i]);

    }

    bank->MAX=a[0]=a[n+1]=-0x7fffffff;

    roof=bank->left=bank->right=bank;

    for (int i=0;i<=n+1;i++) {

       INSERT(i);

       splay(i,roof);

    }

    while (m--) {

        int l,r;

       scanf("%d%d",&l,&r);

       splay(l-1,roof);

       splay(r+1,roof->right);

       printf("%d\n",roof->right->left->MAX);

    }

    fclose(stdin);fclose(stdout);

    return 0;

}

附:以上各代码效率(100000*100000):

fd039245d688d43ffac9f4c87f1ed21b0ef43ba3.jpg.png
5ab5c9ea15ce36d343a774fb3bf33a87e850b15e.jpg.png
58ee3d6d55fbb2fbeee8e5d24e4a20a44723dc2b.jpg.png
0df3d7ca7bcb0a467cd00aef6a63f6246a60af72.jpg.png
09fa513d269759ee43a2e11db3fb43166c22df72.jpg.png
a08b87d6277f9e2f65c79cde1e30e924b999f372.jpg.png

本人乃弱菜一只,以上内容如有不当之处,敬请之处。

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

推荐阅读更多精彩内容