ST算法详解

ST算法是求解RMQ问题的好方法,可以在0(NlogN)的预处理后实现O(1)的查询。该算法是在倍增的思想基础上实现的,比较基础,理解起来也不难。


补充几个要点:

  • RMQ问题:即区间最值问题,给出一个序列a,要求求出区间[l,r]内的最大值。
  • 倍增:(来自lyd的蓝书)
  • log2(x)函数:返回$log_2x$,效率较高,需调用cmath库。
  • 左移运算符(<<):a<<b表示$a*2^b$,效率较高,比乘法运算快。

为了实现O(1)的查询,要先预处理出每个区间的最大值。按照倍增的思想,选取2的非负整数次幂作为区间的边界,然后通过这些区间进行最值的计算。因此不妨用$f_{i,j}$表示区间[i,i+$2^j$-1]的最大值。这样就很明显了,算法过程用递推来实现。

预处理:

显然,区间[i,i+$2^{j-1}$-1]和[i+$2^{j-1}$,i+$2^j$-1]一定覆盖了区间[i,i+$2^j$-1],如下图:

因此,区间[i,i+$2^j$-1]内的最大值就是区间[i,i+$2^{j-1}$-1]和[i+$2^{j-1}$,i+$2^j$-1]内的最大值中更大的一个。可以得出递推式:

f[i][j]=max(f[i][j-1],f[i+(1<<(j-1))][j-1]);

显然,递推边界为$f_{i,0}==a_i$,这个很容易证明。

在预处理的循环过程中,要注意循环边界,以免越界。

预处理代码:

void pre()
{
    int t=log2(n);
    for(int j=1;j<=t;j++)
        for(int i=1;i+(1<<j)-1<=n;i++)
            f[i][j]=max(f[i][j-1],f[i+(1<<(j-1))][j-1]);
}

查询:

先计算出一个满足$2^t<r-l+1<2^{t+1}$的t值,即小于区间长度的2的最高次幂。

显然,区间[l,l+$2^{t}$-1]和[r-$2^t$+1,r]一定覆盖了区间[l,r],如下图:

这个很好证明:因为两个子区间长度均为$2^t$,而区间[l,r]长度小于等于$2^{t+1}$,即$2*2^t$,所以区间[l,l+$2^{t}$-1]和[r-$2^t$+1,r]一定覆盖了区间[l,r]。

因此,区间[l,r]内的最大值就是区间[r-$2^t$+1,r]和[l,l+$2^{t}$-1]内的最大值中更大的一个。可以得出递推式:

ans[l][r]=max(f[l][t],f[r-(1<<t)+1][t]);

在代码实现过程中,可以不定义ans数组,直接输出答案即可。

查询代码:

int calm(int l,int r)
{
    int t=log2(r-l+1);
    return max(f[l][t],f[r-(1<<t)+1][t]);
}

关于f数组的大小,从上面的讲解中应该很好推出:设序列长度为N,则定义f[N][$log_2N$]。数组的大小应该在这个基础上稍大一些,防止出现一些玄学问题。

完整代码:

#include<iostream>
#include<cstdio>
#include<cmath>
using namespace std;
const int N=2e5;
int cn,quel,quer,n,m,f[N][20];
void pre()
{
    int t=log2(n);
    for(int j=1;j<=t;j++)
        for(int i=1;i+(1<<j)-1<=n;i++)
            f[i][j]=max(f[i][j-1],f[i+(1<<(j-1))][j-1]);
}//预处理
int calm(int l,int r)
{
    int t=log2(r-l+1);
    return max(f[l][t],f[r-(1<<t)+1][t]);
}//查询
int main()
{
    cin>>n>>m;
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&cn);
        f[i][0]=cn;
    }//输入
    pre();
    while(m--)
    {
        scanf("%d%d",&quel,&quer);
        printf("%d\n",calm(quel,quer));
    }//在线查询并输出
    return 0;
}

习题:


2019.4.6 于厦门外国语学校石狮分校

posted on 2019-08-20 13:58  TEoS  阅读(1029)  评论(0编辑  收藏  举报