RMQ(Range Minimum/Maximum Query)问题:
RMQ问题是求给定区间中的最值问题。当然,最简单的算法是O(n)的,但是对于查询次数很多(设置多大100万次),O(n)的算法效率不够。可以用线段树将算法优化到O(logn)(在线段树中保存线段的最值)。不过,Sparse_Table算法才是最好的:它可以在O(nlogn)的预处理以后实现O(1)的查询效率。下面把Sparse Table算法分成预处理和查询两部分来说明(以求最小值例)。
令d(i,j)表示从i开始的,长度为2^j的一段元素中的最小值,则可以用递推的方法计算d(i,j):d[i,j]=min{d[i,j-1],d[i+2^(j-1),j-1]},(这个结论很重要,希望着重理解)
注意2^j<=n,因此d数组的元素个数不超过nlogn,而每一项都可以在常数时间计算完成,故总时间为(nlogn).
预处理代码为:
int RMQ_init()
{
int i,j;
for(i = 1;i <= N;i++) d[i][0] = A[i];
for(j = 1;(1<<j) <= N;j++)
for(i = 1;i+(1<<(j-1)) <= N;i++)
d[i][j] = min(d[i][j-1],d[i+(1<<(j-1)][j-1]));
}
假设要查询从R到L这一段的最小值, 首先令 k 为满足 2^k <= R-L+1 的最大整数,则以 L 开头、以 R 结尾的两个长度为 2^k 的区间合起来即覆盖了查询区间 [L, R]。
两个重叠的区间区间 f(L, k) 也就是【L, L+2^k-1】 ,f(R-2^k+1, k) 也就是【R-2^k+1, R】。
由于是取最小值,有些元素重复考虑了几遍也没有关系(注意:如果是累加,重复元素是不许的)。
我们之前已经求出了f(L, k)为[L, L+2^k-1]的最小值, f(R-2^k+1, k)为[R-2^k+1, R]的最小值
我们只要返回其中更小的那个, 就是我们想要的答案, 这个算法的时间复杂度是O(1)的.
例如, RMQ(0, 11) = min(f(0, 3), f(4, 3))
由此我们要注意的是预处理f(i,j)中的j值只需要计算log(n+1)/log(2)即可,而i值我们也只需要计算到n-2^k+1即可。
int RMQ_Min(int L, int R)
{
int k = 0;
while((1<<(k+1)) <= R-L+1) k++; // 如果 2^(k+1) <= R-L+1,那么k 还可以加 1
return min(d_min[L][k], d_min[R-(1<<k)+1][k]);
}
下面推荐几题练手题:
NYOJ 士兵杀敌(三)
POJ Balanced Lineup
下面给出NYOJ的代码吧,跟poj的基本一样
#include <stdio.h>
#include <ctype.h>
#define min(a,b)(a < b ? a : b)
#define max(a,b)(a > b ? a : b)
int minv,maxv;
const int MAX = 100000 + 5;
int d_min[MAX][20],d_max[MAX][20],A[MAX];
int Input()
{
char ch = getchar();
while(!isdigit(ch)) ch = getchar();
int x = 0;
while(isdigit(ch))
{
x = x*10 + ch -'0';
ch = getchar();
}
return x;
}
void RMQ_init(int N)
{
int i,j;
for(i = 1;i <= N;i++)
{
d_min[i][0] = A[i];
d_max[i][0] = A[i];
}
for(j = 1;(1<<j) <= N;j++)
for(i = 1;i+(1<<(j-1))<= N;i++)
{
d_min[i][j] = min(d_min[i][j-1],d_min[i+(1<<(j-1))][j-1]);
d_max[i][j] = max(d_max[i][j-1],d_max[i+(1<<(j-1))][j-1]);
}
}
void RMQ(int L,int R)
{
int k = 0;
while((1<<(k+1)) <= R-L+1) k++;
minv = min(d_min[L][k],d_min[R-(1<<k)+1][k]);
maxv = max(d_max[L][k],d_max[R-(1<<k)+1][k]);
}
int main()
{
//freopen("Input.txt","r",stdin);
int N,Q;
N = Input(); Q = Input(); //printf("N=%d Q=%d\n",N,Q);
for(int i = 1;i <= N;i++)
A[i] = Input();
RMQ_init(N);
int L,R;
while(Q--)
{
L = Input();
R = Input();
RMQ(L,R);
printf("%d\n",maxv - minv);
}
return 0;
}