RMQ——ST算法求解
一.简介
RMQ即询问静态区间最值问题,采用ST算法求解。相比于线段树,更为快捷,但没有修改操作
二.算法思想
基于动态规划,二分与打表算法。先进行预处理(本质为通过动态规划求解所有状态进行打表),O(nlog)完成,再进行O(1)查询。
三.前提知识
位运算a<<b表示a的二进制数左移b位,低位补零。即a*pow(2,b),故1<<j可以表示pow(2,j).
四.算法流程
1.预处理
用a[1…n]表示数列,f[i][j]表示从i出发长度为pow(2,j)的区间中的最大值(或最小值)。由于区间长度为pow(2,j),可以采用二分思想,将区间从中间分成两部分,每一部分区间长度恰好为pow(2,j-1)个,也就是说,把f[i][j]分为f[i][j-1]和f[i+pow(2,j-1)][j-1]。
整个区间的最大值为左右两个小区间的最大值的较大值,因此可得状态转移方程为f[i][j]=max(f[i][j-1],f[i+pow(2,j-1)][j-1])。
初始化为f[i][0]=0,即每个长度为1的区间最大值为自身。
void ST(int n){
for(int j=1;(1<<j)<=n;j++){//枚举所有的区间长度 pow(2,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.询问
若要询问区间[l,r]的最大值,可知,所有状态f[i][j]的区间长度不一定就恰好等于[l][r]的区间长度,f[i][j]所包含的区间可能包含[l][r],也可能被[l][r]所包含。
所以,可以先求出最大的k满足pow(2,k)<=r-l+1,那么[l][r]=[l,l+pow(2,k)-1]∪[r,-pow(2,k)+1,r]。
k=(int)(log(r-l+1)/log(2))
则[l,r]的最大值ans=max(f[l][k],f[r-pow(2,k)+1][k].
int RMQ(int l,int r){
int k=(int)(log(r-l+1)/log(2));//满足pow(2,k)小于等于区间长度的最大k
return max(f[l][k],f[r-(1<<k)+1][k]);//找出[l,r]之间的最大值
}
五.代码模板
#include<bits/stdc++.h>
using namespace std;
int f[100001][65];//表示从i开始长度为pow(2,j)的区间内的最大值
int a[100001];
void ST(int n){
for(int j=1;(1<<j)<=n;j++){//枚举所有的区间长度 pow(2,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 RMQ(int l,int r){
int k=(int)(log(r-l+1)/log(2));//满足pow(2,k)小于等于区间长度的最大k
return max(f[l][k],f[r-(1<<k)+1][k]);//找出[l,r]之间的最大值
}
int main(){
int n,m;
cin>>n>>m;
for(int i=1;i<=n;i++){
scanf("%d",&a[i]);
f[i][0]=a[i];
}
ST(n);
int l,r;
while(m--){
scanf("%d%d",&l,&r);
printf("%d\n",RMQ(l,r));
}
return 0;
}