RMQ算法 (ST算法)

 概述:

  RMQ(Range Minimum/Maximum Query),即区间最值查询,是指这样一个问题:对于长度为n的数列A,回答若干询问RMQ(A,i,j)(i,j<=n),返回数列A中下标在i,j之间的最小/大值。对于一次查询,可以暴力地O(n),但是当查询次数很多的时候,这样的暴力就无法进行了。这时我们可以通过RMQ算法来解决这个问题。

RMQ(ST):(关于学习RMQ的博客:框架即讲解比较详细具体代码比较好

  ST(Sparse Table)算法是一个非常有名的在线处理RMQ问题的算法,它可以在O(nlogn)时间内进行预处理,然后在O(1)时间内回答每个查询。

  首先是预处理,用动态规划(DP)解决。设A[i]是要求区间最值的数列,F[i, j]表示从第i个数起连续2^j个数中的最大值。例如数列3 2 4 5 6 8 1 2 9 7,F[1,0]表示第1个数起,长度为2^0=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]平均分成两段(因为f[i,j]一定是偶数个数字),从i到i+2^(j-1)-1为一段,i+2^(j-1)到i+2^j-1为一段(长度都为2^(j-1))。用上例说明,当i=1,j=3时就是3,2,4,5 和 6,8,1,2这两段。F[i,j]就是这两段的最大值中的最大值。于是我们得到了动态规划方程F[i, j]=max(F[i,j-1], F[i + 2^(j-1),j-1])。

然后是查询。取k=[log2(j-i+1)],则有:RMQ(A, i, j)=min{F[i,k],F[j-2^k+1,k]}。 举例说明,要求区间[2,8]的最大值,就要把它分成[2,5]和[5,8]两个区间,因为这两个区间的最大值我们可以直接由f[2,2]和f[5,2]得到。

 1 int vec[MAX_N];
 2 int dp[MAX_N][25];
 3 void ST(int N)
 4 {
 5     for(int i=1;i<=N;i++) dp[i][0] = vec[i];
 6     for(int j=1;(1<<j) <= N;j++)
 7     {
 8         for(int i=1;i+(1<<j)-1<=N;i++)
 9         {
10             dp1[i][j] = max(dp[i][j-1],dp[i+(1<<j-1)][j-1]); //由于移位操作的优先度低,1<<j-1 = 1<<(j-1);
11         }
12     }
13 }
14 int RMQ(int l,int r)
15 {
16     int k = 0;
17     while((1<<k+1) <= r-l+1) k++;
18     return max(dp1[l][k],dp1[r-(1<<k)+1][k]);
19 }

 

POJ-2364

 1 #include<iostream>
 2 #include<cstring>
 3 #include<cstdio>
 4 using namespace std;
 5 const int MAX_N = 5e4+9;
 6 const int INF = 1e9+7;
 7 int vec[MAX_N];
 8 int dp1[MAX_N][25];
 9 int dp2[MAX_N][25];
10 void ST(int N)
11 {
12     for(int i=1;i<=N;i++) dp1[i][0] = dp2[i][0] = vec[i];
13     for(int j=1;(1<<j)<=N;j++)
14     {
15         for(int i=1;i+(1<<j)-1 <= N;i++)
16         {
17             dp1[i][j] = max(dp1[i][j-1],dp1[i+(1<<j-1)][j-1]);
18             dp2[i][j] = min(dp2[i][j-1],dp2[i+(1<<j-1)][j-1]);
19         }
20     }
21 }
22 int RMQ(int l,int r)
23 {
24     int k = 0;
25     while((1<<k+1) <= r-l+1) k++;
26     return max(dp1[l][k],dp1[r-(1<<k)+1][k]) - min(dp2[l][k],dp2[r-(1<<k)+1][k]);
27 }
28 int main()
29 {
30     int N,M,T;
31     while(cin>>N>>M)
32     {
33         for(int i=1;i<=N;i++)
34         {
35             scanf("%d",&vec[i]);
36         }
37         ST(N);
38         for(int i=0;i<M;i++)
39         {
40             int l,r;
41             scanf("%d%d",&l,&r);
42             int ans = RMQ(l,r);
43             printf("%d\n",ans);
44         }
45     }
46     return 0;
47 }
View Code
posted @ 2018-01-29 09:44  会打架的程序员不是好客服  阅读(215)  评论(0编辑  收藏  举报