RMQ
蒟蒻最近写线段树写自闭了QAQ,就来补一篇RMQ博文。
RMQ (Range Minimum/Maximum Query)问题是指:对于长度为n的数列A,回答若干询问RMQ(A,i,j)(i,j<=n),返回数列A中下标在i,j里的最小(大)值,也就是说,RMQ问题是指求区间最值的问题。
朴素算法
每次询问就一个一个搜,每次查询效率为O(n),本蒟蒻都觉得简单了所以就不写了
线段树
线段树可以做此类问题,不过线段树的编写对于初学者不太友好(本蒟蒻就自闭了)
#include <stdio.h>
#include <string.h>
#define maxn 1000002
#define lson l, mid, rt << 1
#define rson mid + 1, r, rt << 1 | 1
int T[maxn << 2];
int min(int a, int b) {
return a < b ? a : b;
}
void pushUp(int rt) {
T[rt] = min(T[rt<<1], T[rt<<1|1]);
}
void build(int l, int r, int rt) {
if(l == r) {
scanf("%d", &T[rt]);
return;
}
int mid = (l + r) >> 1;
build(lson);
build(rson);
pushUp(rt);
}
void update(int pos, int V, int l, int r, int rt) {
if(l == r) {
T[rt] = V;
return;
}
int mid = (l + r) >> 1;
if(pos <= mid) update(pos, V, lson);
else update(pos, V, rson);
pushUp(rt);
}
int query(int L, int R, int l, int r, int rt) {
if(L == l && R == r) return T[rt];
int mid = (l + r) >> 1;
if(R <= mid) return query(L, R, lson);
else if(L > mid) return query(L, R, rson);
return min(query(L, mid, lson), query(mid + 1, R, rson));
}
int main() {
int N, i, Q, a, c, b;
scanf("%d", &N);
build(1, N, 1);
scanf("%d", &Q);
while(Q--) {
scanf("%d%d%d", &c, &a, &b);
if(c) update(a, b, 1, N, 1);
else printf("%d\n", query(a, b, 1, N, 1));
}
return 0;
}
ST表
ST表是一种处理此类问题优秀的算法,预处理时间 O ( n l o g n ) O(nlogn) O(nlogn),每次查询为 O ( 1 ) O(1) O(1),
ST表实质上是一种动态规划。
设a是要求区间最值的数列,f[i,j]表示从第i个数起连续 2 j 2^j 2j个数中的最大值。
例如数列3 2 4 5 6 8 1 2 9 7 , f [ 1 , 0 ] f[1,0] f[1,0]表示第1个数起,长度为 2 0 = 1 2^0=1 20=1的最大值,其实就是3这个数。f[1,2]=5,f[1,3]=8,f[2,0]=2,f[2,1]=4……从这里可以看出f[i,0]其实就等于a[i]。这样,DP的状态、初值都已经有了,剩下的就是状态转移方程。
我们把f[i,j](j≥1)平均分成两段(因为j≥1时,f[i,j]一定是偶数个数字),从 i i i到 i + 2 j − 1 − 1 i+2^{j-1}-1 i+2j−1−1为一段, i + 2 j − 1 i+2^{j-1} i+2j−1到 i + 2 j − 1 i+2^j-1 i+2j−1为一段(长度都为2^(j-1))。用上例说明,当i=1,j=3时就是3,2,4,5 和6,8,1,2这两段。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])。
那么ST表就建立出来了
查询
接下来是得出最值,也许你想不到计算出f有什么用处,一般要想计算max还是要O(logn),甚至O(n)。
但有一个很好的办法,做到了O(1)。还是分开来。如在上例中我们要求区间[2,8]的最大值,就要把它分成[2,5]和[5,8]两个区间,因为这两个区间的最大值我们可以直接由f[2,2]和f[5,2]得到。
扩展到一般情况,就是把区间[l,r]分成两个长度为 2 n 2^n 2n的区间(这样在f数组中有对应值)。
k
=
t
r
u
n
c
(
l
o
g
2
(
r
−
l
+
1
)
)
;
k=trunc(log_2(r-l+1));
k=trunc(log2(r−l+1));
r
e
s
u
l
t
=
m
a
x
(
F
[
l
,
k
]
,
F
[
r
−
2
k
+
1
,
k
]
)
;
result=max(F[l,k],F[r-2^k+1,k]);
result=max(F[l,k],F[r−2k+1,k]);
这样就计算了从l开始,长度为
2
k
2^k
2k的区间和从r-
2
k
+
1
2^k+1
2k+1开始长度为
2
k
2^k
2k的区间的最大值(表达式比较繁琐,细节问题如加1减1需要仔细考虑),二者中的较大者就是整个区间[l,r]上的最大值。
代码
#include <iostream>
#include <algorithm>
using namespace std;
int a[1010];//原始输入数组
int st[1010][20];//st表
void init(int n)
{
for (int i = 0; i < n; i++)
st[i][0] = a[i];
for (int i = 1; (1 << i) <= n; i++)
{
for (int j = 0; j + (1 << i) - 1 < n; j++)
st[j][i] = min(st[j][i - 1],st[j + (1 << (i - 1))][i - 1]);
}
}
int search(int l, int r)
{
int k = (int)(log((double)(r - l + 1)) / log(2.0));
return min(st[l][k],st[r - (1 << k) + 1][k]);
}
int main()
{
int n,m;
while (cin >> n >> m)
{
for (int i = 0; i < n; i++)
cin >> a[i];
init(n);
while (m--)
{
int l, r;
cin >> l >> r;
cout << search(l,r) << endl;;
}
}
return 0;
}
例题
模板 洛谷P3865
标准解法
大概过程就是RMQ->LCA->约束RMQ
有兴趣者自行了解QAQ,蒟蒻是打算以后再填坑了。