RMQ问题即求解区间最值问题,今天学习一下如何是有ST算法求解RMQ问题。
ST算法可以在O(nlogn)的预处理以后实现O(1)的查询效率,从而解决查询次数较多的RMQ问题。
首先,是预处理。预处理采取的思想是动态规划,我们用f[i][j]表示表示区间[i,]的最大值。
开始的时候,f[i][0]一定要等于num[i]。初始值找到之后,我们接下来寻找状态转移方程:
f[i][j] = max(f[i][j - 1],f[i+2^(j - 1)][j-1])。即把原区间划分为两部分[i,i+2^(j-1)-1]和[i+2^(j-1),i+2^j - 1]正好和原区间一样。
有了初值和转移方程,我们可以自底向上的退出所有f[i][j]的值。
考虑完成初始值和转移方程之后,我们需要思考一下边界问题,由于区间的最大长度为n,因此二维边界最大为log(n)/log(2.0);一维边界只要满足对于每个起点到n都能长度就行,也就是i+2^j-1<=n。
然后就是查询了。假设我们查询的区间为[a,b],由于区间长度(b-a+1)不一定是2的幂,所以我们需要把区间划分为长度为2的幂的两部分,而且这两部分的并集必须是[a,b]。为了实现这个方法,我么需要求出一个最大的K使得2^k<=(b-a+1),这样我们可以把区间划分为[a,a + 2^k -1]和[a + 2^k,b],使它们即不超过[a,b]这个区间,又能够把[a,b]这个区间覆盖。[a,b]区间的最大值就为上述两个区间中的最大值中的最大值。
示例代码如下,如有错误请予以指出。
/*这里只展示区间最大值的代码,最小值只需要改变比较函数即可*/
#include <iostream>
#include <cmath>
#include <cstdio>
using namespace std;
const int maxn = 10005;
int n,q,num[maxn];
int dp[maxn][20];//20需要根据log(n)/log(2.0)得出
void initialize()
{
for(int i = 1; i<=n; i++)
dp[i][0]=num[i];
int limt = (int)(log(n*1.0)/log(2.0));
for(int j = 1; j<=limt; j++)
{
for(int i = 1; i<=n; i++)
{
dp[i][j] = dp[i][j-1];
if(i + (1<<(j-1)) <= n)
dp[i][j] = max(dp[i][j],dp[i + (1<<(j-1))][j-1]);
}
}
}
int getMax(int a,int b)
{
int k = (int)(log((b-a+1)*1.0)/log(2.0));
return max(dp[a][k],dp[b-(1<<k)+1][k]);
}
int main()
{
scanf("%d %d",&n,&q);
for(int i = 1; i<=n; i++)
scanf("%d",&num[i]);
initialize();
int a,b;
while(q--)
{
scanf("%d%d",&a,&b);
printf("%d\n",getMax(a,b));
}
return 0;
}