RMQ
1. RMQ原理
原理
-
RMQ
算法全称为Range Minimum/Maximum Query
,即用来解决区间最值问题。这类问题可以使用线段树解决,但是代码比较长。这里演示使用RMQ
算法解决,其实就是动态规划。 -
使用
f(i, j)
表示:从i
开始长度为 2 j 2^j 2j的区间中的最大值。 -
求解
f
的递推公式: f ( i , j ) = m a x ( f ( i , j − 1 ) , f ( i + 2 j − 1 , j − 1 ) ) f(i, j) = max(f(i, j-1), f(i+2^{j-1}, j-1)) f(i,j)=max(f(i,j−1),f(i+2j−1,j−1))。 -
预处理出来
f
之后,对于给定查询[l, r]
之间的最大值,记len=r-l+1
,我们需要找到最大的k
使得 2 k ≤ l e n 2^k \le len 2k≤len,则区间最大值为 m a x ( f ( l , k ) , f ( r − 2 k + 1 , k ) ) max(f(l, k), f(r-2^k+1, k)) max(f(l,k),f(r−2k+1,k))。 -
这里的
k
的求解可以使用log
函数(c++
的log
函数默认以e
为底,需要使用换底公式)。
2. AcWing上的RMQ题目
AcWing 1270. 数列区间最大值
问题描述
- 问题链接:AcWing 1270. 数列区间最大值
分析
-
直接使用RMQ算法即可。
-
这里区间长度最大为
100000
,由于 2 16 = 65526 2^{16}=65526 216=65526,因此j
最大取16
即可( 2 16 2^{16} 216是最近接100000
的2
的指数),因此数组第二维是17
。
代码
#include <iostream>
#include <cmath>
using namespace std;
const int N = 100010, M = 17;
int n, m;
int w[N];
int f[N][M];
void init() {
for (int j = 0; j < M; j++)
for (int i = 1; i + (1 << j) - 1 <= n; i++)
if (!j) f[i][j] = w[i];
else f[i][j] = max(f[i][j - 1], f[i + (1 << j - 1)][j - 1]);
}
int query(int l, int r) {
int len = r - l + 1;
int k = log(len) / log(2);
return max(f[l][k], f[r - (1 << k) + 1][k]);
}
int main() {
ios::sync_with_stdio(false);
cin.tie(0);
cin >> n >> m;
for (int i = 1; i <= n; i++) cin >> w[i];
init();
while (m--) {
int a, b;
cin >> a >> b;
cout << query(a, b) << endl;
}
return 0;
}
AcWing 1273. 天才的记忆
问题描述
-
问题链接:AcWing 1273. 天才的记忆
分析
-
直接使用RMQ算法即可。
-
这里区间长度最大为
200000
,由于 2 16 = 65526 2^{16}=65526 216=65526,因此j
最大取17
即可( 2 17 2^{17} 217是最近接200000
的2
的指数),因此数组第二维是18
。
代码
- C++
#include <iostream>
#include <cmath>
using namespace std;
const int N = 200010, M = 18;
int n, m;
int w[N];
int f[N][M];
void init() {
for (int j = 0; j < M; j++)
for (int i = 1; i + (1 << j) - 1 <= n; i++)
if (!j) f[i][j] = w[i];
else f[i][j] = max(f[i][j - 1], f[i + (1 << j - 1)][j - 1]);
}
int query(int l, int r) {
int len = r - l + 1;
int k = log(len) / log(2);
return max(f[l][k], f[r - (1 << k) + 1][k]);
}
int main() {
scanf("%d", &n);
for (int i = 1; i <= n; i++) scanf("%d", &w[i]);
init(); // 计算数组f
scanf("%d", &m);
while (m--) {
int l, r;
scanf("%d%d", &l, &r);
printf("%d\n", query(l, r));
}
return 0;
}