一、RMQ问题描述
RMQ (Range Minimum/Maximum Query)问题是指:对于长度为n的数列A,回答若干询问RMQ(A,i,j)(i,j<=n),返回数列A中下标在[i,j]里的最小(大)值,也就是说,RMQ问题是指求区间最值的问题
ST算法(Sparse Table):它是一种动态规划的方法。
以最小值为例。a为所寻找的数组.
用一个二维数组dp(i,j)记录区间[i,i+2^j-1](持续2^j个)区间中的最小值。其中dp[i,0] = a[i]; 所以,对于任意的一组(i,j),dp(i,j) = min{dp(i,j-1),dp(i+2^(j-1),j-1)}来使用动态规划计算出来。
这个算法的高明之处不是在于这个动态规划的建立,而是它的查询:它的查询效率是O(log(j-i)). 假设我们要求区间[m,n]中a的最小值,找到一个数k使得2^k<n-m+1)并且2^(k+1)>=n-m+1.
这样,可以把这个区间分成两个部分:[m,m+2^k-1]和[n-2^k+1,n].这两个部分要一定保证完全覆盖这个区间,我们发现,这两个区间是已经初始化好的.
前面的区间是dp(m,k),后面的区间是dp(n-2^k+1,k).
这样,只要看这两个区间的最小值,就可以知道整个区间的最小值!
二、RMQ的应用
1、poj1207
/*裸的RMQ */ #include <iostream> #include <cstdio> #include <cstring> using namespace std; #define max(a,b) ((a)>(b)?(a):(b)) int dat[10002]; int dp[10002][18]; int get_dat(int n) { int cnt=1; while(n!=1) { if(n&1){n=3*n+1;} else{n/=2;} ++cnt; } return cnt; } void make_max_rmq() { for(int j=1;(1<<j)<=10000;++j) { for(int i=1;i+(1<<j)-1<=10000;++i) { dp[i][j]=max(dp[i][j-1],dp[i+(1<<(j-1))][j-1]); } } } int get_max_rmq(int a,int b) { int k=0; while(a+(1<<k)<b-(1<<k)+1)++k; return max(dp[a][k],dp[b-(1<<k)+1][k]); } int main() { for(int i=1;i<=10000;++i) { dat[i]=get_dat(i); dp[i][0]=dat[i]; } int l,h,a,b; make_max_rmq(); while(~scanf("%d%d",&a,&b)) { l=a<b?a:b; h=a>b?a:b; int ans=get_max_rmq(l,h); printf("%d %d %d\n",a,b,ans); } return 0; }
2、poj3264
/*n个数,求i,j的最大值和最小值之差 */ #include <iostream> #include <cstdio> #include <cstring> using namespace std; #define max(a,b) ((a)>(b)?(a):(b)) #define min(a,b) ((a)<(b)?(a):(b)) int m,n; int dat[50002]; int dpmax[50002][18]; int dpmin[50002][18]; void make_max_rmq() { for(int i=1;i<=n;++i) { dpmax[i][0]=dat[i]; } for(int j=1;(1<<j)<=n;++j) { for(int i=1;i+(1<<j)-1<=n;++i) { dpmax[i][j]=max(dpmax[i][j-1],dpmax[i+(1<<(j-1))][j-1]); } } } void make_min_rmq() { for(int i=1;i<=n;++i) { dpmin[i][0]=dat[i]; } for(int j=1;(1<<j)<=n;++j) { for(int i=1;i+(1<<j)-1<=n;++i) { dpmin[i][j]=min(dpmin[i][j-1],dpmin[i+(1<<(j-1))][j-1]); } } } int get_max_rmq(int a,int b) { int k=0; while(a+(1<<k)<b-(1<<k)+1)++k; return max(dpmax[a][k],dpmax[b-(1<<k)+1][k]); } int get_min_rmq(int a,int b) { int k=0; while(a+(1<<k)<b-(1<<k)+1)++k; return min(dpmin[a][k],dpmin[b-(1<<k)+1][k]); } int main() { int a,b; while(~scanf("%d%d",&n,&m)) { for(int i=1;i<=n;++i) { scanf("%d",&dat[i]); } make_max_rmq(); make_min_rmq(); while(m--) { scanf("%d%d",&a,&b); int ans=get_max_rmq(a,b)-get_min_rmq(a,b); printf("%d\n",ans); } } return 0; }
3、poj3368
/* * 有n个数升序排列,输出i,j的最大的众数 */ #include <iostream> #include <cstdio> #include <cstring> #include <cmath> using namespace std; #define max(a,b) ((a)>(b)?(a):(b)) #define N 100002 int n,m; int rank[N]; int dat[N]; int dp[N][18]; int rear[N]; int head[N]; void make_max_rmq() { for(int i=1;i<=n;++i) { dp[i][0]=rank[i]; } for(int j=1;j<=(int)(log((double)n)/log(2.0));++j) { for(int i=1;i+(1<<j)-1<=n;++i) { dp[i][j]=max(dp[i][j-1],dp[i+(1<<(j-1))][j-1]); } } } int get_max_rmq(int a,int b) { int k=0; while(a+(1<<k)<b-(1<<k)+1)++k; return max(dp[a][k],dp[b-(1<<k)+1][k]); } int main() { int a,b; while(~scanf("%d",&n)) { if(!n)break; scanf("%d",&m); int cnt=0; memset(head,0,sizeof(head)); memset(rear,0,sizeof(rear)); head[0]=1; for(int i=1;i<=n;++i) { scanf("%d",&dat[i]); if(dat[i]==dat[i-1]||i==1) { rank[i]=++cnt; head[i]=head[i-1]; } else{ head[i]=i; rear[head[i-1]]=i-1; rank[i]=1; cnt=1; } } make_max_rmq(); while(m--) { scanf("%d%d",&a,&b); int ans; if(dat[a]==dat[b]){ans=b-a+1;} else if(rear[head[a]]+1==head[b]) { ans=max(rear[head[a]]-a+1,b-head[b]+1); } else { ans=max(rear[head[a]]-a+1,b-head[b]+1); ans=max(ans,get_max_rmq(rear[head[a]]+1,head[b]-1)); } printf("%d\n",ans); } } return 0; }
4、poj2109 二维RMQ
/* 求子矩阵的最大值与最小值 */ #include <iostream> #include <cstdio> #include <cstring> using namespace std; #define max(a,b) ((a)>(b)?(a):(b)) #define min(a,b) ((a)<(b)?(a):(b)) #define inf 0x7ffffff int dpmax[252][252][8]; int dpmin[252][252][8]; int dat[252][252]; int n,b,k; void make_rmq2() { for(int k=1;(1<<k)<=n;++k) { for(int i=1;i<=n;++i) { for(int j=1;j<=n;++j) { dpmax[i][j][k]=dpmax[i][j][k-1]; if(i+(1<<(k-1))<=n) dpmax[i][j][k]=max(dpmax[i][j][k],dpmax[i+(1<<(k-1))][j][k-1]); if(j+(1<<(k-1))<=n) dpmax[i][j][k]=max(dpmax[i][j][k],dpmax[i][j+(1<<(k-1))][k-1]); if((i+(1<<(k-1))<=n)&&(j+(1<<(k-1))<=n)) dpmax[i][j][k]=max(dpmax[i][j][k],dpmax[i+(1<<(k-1))][j+(1<<(k-1))][k-1]); dpmin[i][j][k]=dpmin[i][j][k-1]; if(i+(1<<(k-1))<=n) dpmin[i][j][k]=min(dpmin[i][j][k],dpmin[i+(1<<(k-1))][j][k-1]); if(j+(1<<(k-1))<=n) dpmin[i][j][k]=min(dpmin[i][j][k],dpmin[i][j+(1<<(k-1))][k-1]); if((i+(1<<(k-1))<=n)&&(j+(1<<(k-1))<=n)) dpmin[i][j][k]=min(dpmin[i][j][k],dpmin[i+(1<<(k-1))][j+(1<<(k-1))][k-1]); } } } } int get_max_rmq2(int x,int y,int k) { int r=dpmax[x][y][k]; if(x+b-(1<<k)<=n) r=max(r,dpmax[x+b-(1<<k)][y][k]); if(y+b-(1<<k)<=n) r=max(r,dpmax[x][y+b-(1<<k)][k]); if((x+b-(1<<k)<=n)&&(y+b-(1<<k)<=n)) r=max(r,dpmax[x+b-(1<<k)][y+b-(1<<k)][k]); return r; } int get_min_rmq2(int x,int y,int k) { int r=dpmin[x][y][k]; if(x+b-(1<<k)<=n) r=min(r,dpmin[x+b-(1<<k)][y][k]); if(y+b-(1<<k)<=n) r=min(r,dpmin[x][y+b-(1<<k)][k]); if((x+b-(1<<k)<=n)&&(y+b-(1<<k)<=n)) r=min(r,dpmin[x+b-(1<<k)][y+b-(1<<k)][k]); return r; } int main() { int x,y; while(~scanf("%d%d%d",&n,&b,&k)) { for(int i=1;i<=n;++i) { for(int j=1;j<=n;++j) { scanf("%d",&dat[i][j]); dpmax[i][j][0]=dpmin[i][j][0]=dat[i][j]; } } int p=0; while((1<<(p+1))<=b){++p;} make_rmq2(); while(k--) { scanf("%d%d",&x,&y); int ans=get_max_rmq2(x,y,p)-get_min_rmq2(x,y,p); printf("%d\n",ans); } } return 0; }
5、Poj2452 RMQ+两次二分
/* 题意:有一串N个不同数字的序列,求出一个最长的连续子序列si~sj,满足si是最小数,sj是最大数。 思路:首先预处理,对于每个si,找出以它为起点后面最大数的位置rear[i].然后二分找到以si为最小值的区间,然后在这个区间里面二分寻找最大值的位置j,j-i的最大值即为所求 */ #include <iostream> #include <cstdio> #include <cstring> using namespace std; #define min(a,b) ((a)<(b)?(a):(b)) #define max(a,b) ((a)>(b)?(a):(b))。 const int N=5e4+2; int n; int dat[N]; int dpmax[N][30]; int dpmin[N][30]; int rear[N]; void make_rmq() { for(int j=1;(1<<j)<=n;++j) { for(int i=1;i+(1<<j)-1<=n;++i) { dpmax[i][j]=max(dpmax[i][j-1],dpmax[i+(1<<(j-1))][j-1]); dpmin[i][j]=min(dpmin[i][j-1],dpmin[i+(1<<(j-1))][j-1]); } } } int get_max_rmq(int a,int b) { int k=0; while(a+(1<<k)<b-(1<<k)+1)++k; return max(dpmax[a][k],dpmax[b-(1<<k)+1][k]); } int get_min_rmq(int a,int b) { int k=0; while(a+(1<<k)<b-(1<<k)+1)++k; return min(dpmin[a][k],dpmin[b-(1<<k)+1][k]); } int main() { while(~scanf("%d",&n)) { for(int i=1;i<=n;++i) { scanf("%d",&dat[i]); dpmin[i][0]=dpmax[i][0]=dat[i]; } int t=n; rear[n]=n; for(int i=n-1;i>=1;--i) { if(dat[i]<dat[t]) { rear[i]=t; } else { rear[i]=i; t=i; } } for(int i=1;i<=n;++i) { } int ans=-1; make_rmq(); for(int i=1;i<=n;++i) { int low=i,high=rear[i],mid,t=-1; if(get_min_rmq(i,high)==dat[i]){t=max(t,high);} else{ while(low<high) { if(get_min_rmq(i,low)==dat[i]) {t=max(t,low);} if(get_min_rmq(i,high)==dat[i]) {t=max(t,high);} mid=(low+high)/2; if(get_min_rmq(i,mid)==dat[i]) { t=max(t,mid); low=mid+1; } else{high=mid-1;} } } high=t;low=i; int w=get_max_rmq(low,high); if(get_max_rmq(i,high)==dat[high]) { ans=max(ans,high-i); } else { while(low<high) { if(dat[low]==w) {ans=max(ans,low-i);} if(dat[high]==w) {ans=max(ans,high-i);} mid=(low+high)/2; if(get_max_rmq(i,mid)<w) {low=mid+1;} else{high=mid-1; if(dat[mid]==w)ans=max(ans,mid-i); } } } } if(ans==0)ans=-1; printf("%d\n",ans); } return 0; }