RMQ问题ST算法与模板(2007-07-15 15:48)

 

-------------------------------------算法简述-----------------------------------------

ST算法O(nlogn)预处理,O(1)的查询指定区间的最值(以最小值为例)

基本上是把待求区间[l,r]分为两段长为len的区间

左边一段为[l,l+len-1],右边一段为[r-len+1,r]

len必须使得两段区间覆盖待求区间

设所求数组为w

那么,所求最小值就是两个区间的最小值间的最小值

即min(min{w[i],l<=i<=l+len-1},min{w[j],r-len+1<=j<=r})

若都在预先处理中先求得两个区间的最小值

则每次查询的复杂度都是O(1)

---

对len做一个限制:只能为2的幂

在预处理中求出所有mi[b][t] : 以b为起点,长为2^t的区间的最小值.

则求解min(min{w[i],l<=i<=l+len-1},min{w[j],r-len+1<=j<=r})

就变成min(mi[l][t],mi[r-2^t+1][r]),其中t可以由此得出,以保证两段区间可以覆盖待求区间:

t=ln(r-l+1)/ln(2)

---

可以看到mi[b][t]=min(mi[b][t-1],mi[b+2^(t-1)-1][t-1])

特别地对于所有mi[i][0],其值都是w[i];

由此自底向上推出所有的mi[b][t]

mi大小为n*logn,预处理时间复杂度为O(nlogn),查询时间复杂度为O(1)

-----------------------------------------------------代码------------------------------------------------------

在POJ上测试, 反应良好~

不知道怎么样把函数指针指向任意类型的<操作符, 只能总是提供一个"小于"函数

查询指定区间的最小值.

----------------------------------以下是模板-------------------------------------

 

1. template < class Type>
2. class rmq_st{
3. private:
4.       Type* body;   
5.       int** mi;      
6.       int size;     
7.       int len;     
9.       int dint(double a){       
10.           int buf=a;   
11.           if(buf>a){
12.               --buf;
13.           }
14.           return buf;
15.       }
17.       int mlen(int l,int r){
18.           double buf=log((double)(r-l+1))/log((double)2);  
19.           return dint(buf)
20.       }   
21.       bool (*less)(Type& t1,Type& t2);
23. public:
26.      //构造函数 : (待求数组,大小,小于函数)      
27.       rmq_st(Type* con,int s,bool (*lessthan)(Type& t1,Type& t2));
28.            
29.       ~rmq_st();//解构
30.      
31.       Type get_body(int p);//返回指定索引的元素
33.       int query_index(int l,int r);//返回指定区间最小值的元素索引
35.       Type query(int l,int r);//返回指定区间的最小元素
36. };
38. template < class Type> 
39. rmq_st< Type>::rmq_st(Type* con,int s,bool (*lessthan)(Type& t1,Type& t2)){
40.       less=lessthan;
41.       body=con;
42.       size=s;
43.       len=mlen(0,s-1)+1;
44.       mi=new int*[size];
45.       int i,j;
46.       for(i=0;i< size;++i){
47.           mi[i]=new int[len];
48.       }
49.       int bound;  
50.       int a,b;    
51.       for(i=0;i< size;++i){   
52.           mi[i][0]=i; 
53.       } 
54.       for(i=1;i< len;++i){     
55.           bound=n-(1<< i)+1;      
56.           for(j=0;j< bound;++j){      
57.               a=mi[j][i-1];       
58.               b=mi[j+(1<< (i-1))][i-1];    
59.               mi[j][i]=less(body[a],body[b])?a:b;
60.           } 
61.       }
62. }
64. template < class Type> 
65. rmq_st< Type>::~rmq_st(){
66.       int i;
67.       for(i=0;i< size;++i){
68.           delete[] mi[i];
69.       }    
70.       delete[] mi;
71. }
73. template < class Type> 
74. Type rmq_st< Type>::get_body(int p){
75.       return body[p];
76. }
78. template < class Type> 
79. int rmq_st< Type>::query_index(int l,int r){
80.       int length=mlen(l,r);        
81.       int a=mi[l][length];
82.       int b=mi[r-(1<< length)+1][length];
83.       return less(body[a],body[b])?a:b;
84. }
86. template < class Type> 
87. Type rmq_st< Type>::query(int l,int r){
88.       int length=mlen(l,r);
89.       int a=mi[l][length];
90.       int b=mi[r-(1<< length)+1][length];
91.       return less(body[a],body[b])?body[a]:body[b];
92. }

/*----------------Example : POJ 3264    Memory:7708K   Time:3230MS-----------------*/

1. int cow[50001];
2. int n,q;
4. bool cmpmin(int& t1,int& t2){
5.       return t1< t2;
6. }
8. bool cmpmax(int& t1,int& t2){
9.       return t1>t2;
10. }
12. int main()
13. {
14. //    freopen("1.in","r",stdin);
15.       scanf("%d %d",&n,&q);
16.       int i;    
17.       for(i=0;i< n;++i){
18.           scanf("%d",&cow[i]);
19.       }
20.       rmq_st< int> minst(cow,n,cmpmin);
21.       rmq_st< int> maxst(cow,n,cmpmax);
22.       int a,b;
23.       for(i=0;i< q;++i){
24.           scanf("%d %d",&a,&b);
25.           --a;
26.           --b;
27.           printf("%d/n",maxst.query(a,b)-minst.query(a,b));
28.       }      
29.       return 0;
30. }