RMQ问题,就是求给定区间内最大值和最小值的问题。RMQ问题有很多种解法,例如线段树。这次写的是ST算法,一种动归算法。实际上每个人都能写出来,但就是不会推DP转移方程,导致了这个算法模板的出现。
给出一段区间,然后去求这个区间的最大值和最小值,可能最先想到的算法是线段树,毕竟线段树是维护区间的最常用的算法,而线段树的效率在一些题目中却是刚好被几个时间点卡死的。所以就出现了时间复杂度更低的ST算法。
maxsum[i][j] = max(maxsum[i][j - 1], maxsum[i + (1 << (j - 1))][j - 1]); minsum[i][j] = min(minsum[i][j - 1], minsum[i + (1 << (j - 1))][j - 1]);
先给出DP转移方程,可能这个方程对初学者来言很奇怪,根本看不懂什么意思。实际上maxsum和minsum中是指从第i位开始,向后长度为2^j - 1的区间内,最大值和最小值分别是多少。这样一说以后,这个dp方程就显得很简单了,这个方程明显透露了一个思想,就是我们先1和2比较,2和3比较,3和4比较……然后在两两看做一个整体的区间,再分别去比较,最终求解我们想要的答案。(感觉上跟二分然后再回来查询差不多)
至于上面的DP方程,实际上是预处理中完成的,其时间复杂度为O(nlgn),而实际上的查询操作是离线的O(1)查询,这便使得ST算法的效率比起线段树来高了不少,于是查询核心代码如下
scanf("%d %d", &src, &des); int k = (int)(log(des - src + 1.0) / log(2.0)); int maxres = max(maxsum[src][k], maxsum[des - (1 << k) + 1][k]); int minres = min(minsum[src][k], minsum[des - (1 << k) + 1][k]); printf("%d\n", maxres - minres);
其中,查询时就是搜索以SRC开始的长度为2^k区间和以des为终结点长度为2^K的区间,比较两个区间的最大值和最小值,由于已经初始化过,所以查询只需要O(1)的时间复杂度了。
于是,一篇成型的ST算法就此出炉了……只不过这里面需要用到log函数,请大家自己学习对数函数的知识,这里用到的只不过是一个小小的换底公式
于是给出整篇成型的ST算法如下,以poj 3264 Balanced Lineup为例:
1 #include<cstdio> 2 #include<algorithm> 3 #include<cmath> 4 using namespace std; 5 6 const int N = 100010; 7 int maxsum[N][20], minsum[N][20]; 8 9 void RMQ(int num) //预处理->O(nlogn) 10 { 11 for(int j = 1; j < 20; ++j) 12 for(int i = 1; i <= num; ++i) 13 if(i + (1 << j) - 1 <= num) 14 { 15 maxsum[i][j] = max(maxsum[i][j - 1], maxsum[i + (1 << (j - 1))][j - 1]); 16 minsum[i][j] = min(minsum[i][j - 1], minsum[i + (1 << (j - 1))][j - 1]); 17 } 18 } 19 20 int main() 21 { 22 int num, query; 23 int src, des; 24 scanf("%d %d", &num, &query); 25 for(int i = 1; i <= num; ++i) 26 { 27 scanf("%d", &maxsum[i][0]); 28 minsum[i][0] = maxsum[i][0]; 29 } 30 RMQ(num); 31 while(query--) 32 { 33 scanf("%d %d", &src, &des); 34 int k = (int)(log(des - src + 1.0) / log(2.0)); 35 int maxres = max(maxsum[src][k], maxsum[des - (1 << k) + 1][k]); 36 int minres = min(minsum[src][k], minsum[des - (1 << k) + 1][k]); 37 printf("%d\n", maxres - minres); 38 } 39 return 0; 40 }
ST算法实际上就是一个dp算法,只要对dp理解的够深,就有可能自己写出ST甚至是更加有效地算法。