RMQ(Range Minimum/Maximum Query)问题:
RMQ问题是求给定区间中的最值问题。当然,最简单的算法是O(n)的,但是对于查询次数很多(设置多大100万次),O(n)的算法效率不够。可以用线段树将算法优化到O(logn)(在线段树中保存线段的最值)。不过,Sparse_Table算法才是最好的:它可以在O(nlogn)的预处理以后实现O(1)的查询效率。下面把Sparse Table算法分成预处理和查询两部分来说明(以求最小值为例)。
预处理:
预处理使用DP的思想,f(i, j)表示[i, i+2^j - 1]区间中的最小值,我们可以开辟一个数组专门来保存f(i, j)的值。
例如,f(0, 0)表示[0,0]之间的最小值,就是num[0], f(0, 2)表示[0, 3]之间的最小值, f(2, 4)表示[2, 17]之间的最小值
注意, 因为f(i, j)可以由f(i, j - 1)和f(i+2^(j-1), j-1)导出, 而递推的初值(所有的f(i, 0) = i)都是已知的
所以我们可以采用自底向上的算法递推地给出所有符合条件的f(i, j)的值。
查询:
假设要查询从m到n这一段的最小值, 那么我们先求出一个最大的k, 使得k满足2^k <(n - m + 1).
于是我们就可以把[m, n]分成两个(部分重叠的)长度为2^k的区间: [m, m+2^k-1], [n-2^k+1, n];
而我们之前已经求出了f(m, k)为[m, m+2^k-1]的最小值, f(n-2^k+1, k)为[n-2^k+1, n]的最小值
我们只要返回其中更小的那个, 就是我们想要的答案, 这个算法的时间复杂度是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即可。
#include <cstdio>
#include <algorithm>
using namespace std;
int dp1[50000+10][20], dp2[50000+10][20];
int a[50000+10];
int main()
{
int n, q, i, j;
while(scanf("%d %d", &n, &q) == 2)
{
for(i=1; i<=n; i++)
{
scanf("%d", &a[i]);
dp1[i][0] = a[i];
dp2[i][0] = a[i];
}
for(j=1; j<=20; j++)
for(i=1; i<n; i++)
{
if(i+(1<<j)-1 > n)
break;
dp1[i][j] = max(dp1[i][j-1], dp1[i+(1<<j-1)][j-1]);
dp2[i][j] = min(dp2[i][j-1], dp2[i+(1<<j-1)][j-1]);
}
while(q--)
{
int l, r;
scanf("%d %d", &l, &r);
for(i=1; i<=20; i++)
if(l+(1<<i)-1 > r)
break;
for(j=1; j<=20; j++)
if(r-(1<<j)+1 < l)
break;
int k1 = max(dp1[l][i-1], dp1[r-(1<<j-1)+1][j-1]);
int k2 = min(dp2[l][i-1], dp2[r-(1<<j-1)+1][j-1]);
printf("%d\n", k1 - k2);
}
}
return 0;
}
精简后的代码
使用位运算可以加快速度,之前做过的提回顾一下还是蛮好的
#include <cstdio>
#include <algorithm>
using namespace std;
int dp1[20][50010], dp2[20][50010];
int a[50010], LOG[50010];
int main()
{
int n, q, i, j;
LOG[1] = 0;
for(i=2; i<=50010; i++)
LOG[i] = LOG[i>>1] + 1;
while(scanf("%d %d", &n, &q) == 2)
{
for(i=1; i<=n; i++)
{
scanf("%d", &a[i]);
dp1[0][i] = a[i];
dp2[0][i] = a[i];
}
for(i=1; i<=20; i++)
for(j=1; j<=50000; j++)
{
if(j+(1<<i)-1 > n)
break;
dp1[i][j] = max(dp1[i-1][j], dp1[i-1][j+(1<<i>>1)]);
dp2[i][j] = min(dp2[i-1][j], dp2[i-1][j+(1<<i>>1)]);
}
while(q--)
{
int l, r;
scanf("%d %d", &l, &r);
i = LOG[r-l+1];
int k1 = max(dp1[i][l], dp1[i][r-(1<<i)+1]);
int k2 = min(dp2[i][l], dp2[i][r-(1<<i)+1]);
printf("%d\n", k1 - k2);
}
}
return 0;
}