问题定义:
RMQ(Range Maximum/Minimum Query)的缩写,顾名思义是用来求某个区间内的最大值或最小值,通常用在需要多次询问一些区间的最值的问题中。
原理:
用val[1..N]表示一组数,F[i,j]表示从val[i]到val[i+2^j-1]这个范围内的最大值,也就是以val[i]为起点连续2^j个数的最大值,由于元素个数为2^j个,所以从中间平均分成两部分,每一部分的元素个数刚好为2^(j-1),如下图。
整个区间的最大值一定是左右两部分最大值的较大值,满足动态规划的最优原理
状态转移方程:
F[i, j] = max { F[i, j-1] , F[i+2^(j-1), j-1] }
边界条件为F[i, 0] = val[i]
查询:
对于询问[L, R],求出最大的x,满足2^x <= R-L+1,即x=ln(R-L+1)/ln2
[L, R] = [L, L+2^x-1] ∪ [R+1-2^x, R],两个子区间元素个数都是2^x个。如图
ANS(L, R) = max { F[L,x], F[R+1-2^x, x] }
代码模板:
#define SIZE 50005
int rmqMax[SIZE][20], rmqMin[SIZE][20]; // 储存动规结果
int val[SIZE]; // 原始数值
// n表示val[]的实际长度
void InitRMQ(int n) {
for (int i = 1; i <= n; ++ i)
rmqMax[i][0] = rmqMin[i][0] = i;
int len = (int)(log(n*1.0)/log(2.0));
for (int j = 1; j <= len; ++ j) {
for (int i = 1; i <= n+1-1<<j; ++ i) {
int x1 = rmqMin[i][j-1];
int y1 = rmqMin[i+(1<<(j-1))][j-1];
int x2 = rmqMax[i][j-1];
int y2 = rmqMax[i+(1<<(j-1))][j-1];
rmqMin[i][j] = val[x1]<val[y1]?x1:y1;
rmqMax[i][j] = val[x2]>val[y2]?x2:y2;
}
}
};
// 2^x <= R-L+1
// query_max(L,R) = max { F[L, x], F[R+1-2^x, x] }
int query_max(int l, int r) {
int x = (int)(log(r-l+1.0)/log(2.0));
int a = rmqMax[l][x];
int b = rmqMax[r+1-(1<<x)][x];
return val[a]>val[b]?val[a]:val[b];
};
int query_min(int l, int r) {
int x = (int)(log(r-l+1.0)/log(2.0));
int a = rmqMin[l][x];
int b = rmqMin[r+1-(1<<x)][x];
return val[a]<val[b]?val[a]:val[b];
};