您的位置:首页 > 理论基础 > 数据结构算法

[数据结构-查询区间最小值小结(RMQ问题(Range Minimum Query))]

2016-03-14 15:44 267 查看
查询区间第k大可以用划分树,而查询区间最小(最大)值可以用代码更简单的Sparse-Table(ST)算法(Tarjan发明)。

原理很简单,代码也很短:

令d(i,j)表示从i开始的,长度为2^j的一段元素最小值,那么可以用递推方法计算:

d(i,j) = min( d(i,j-1) , d(i+2^(j-1) , j-1) )

void RMQ_init(const vector<int>& A)  {
int n=A.size();
for(int i=0;i<n;i++) d[i][0]=A[i];
for(int j=1;(1<<j)<=n;j++)
for(int i=0;i+(1<<j)-1<n;i++)
d[i][j]=min(d[i][j-1],d[i+(1<<(j-1))][j-1]);
}


查询操作也很简单。查询区间[L,R]内最小值:令k为满足2^k<=R-L+1的最大整数,则以L开头、以R结尾的两个长度为2^k的区间合起来就覆盖了整个[L,R]区间。

int RMQ(int L,int R) {
int k=0;
while((1<<(k+1))<=R-L+1) k++;
return min(d[L][k],d[R-(1<<k)+1][k]);
}


 

 

下面的代码预处理了mm数组,它表示k的值,所以速度更快:

const int MAXN = 50010;
int dp[MAXN][20];
int mm[MAXN];
//初始化RMQ, b数组下标从1开始,从0开始简单修改
void initRMQ(int n,int b[])
{
mm[0] = -1;
for(int i = 1; i <= n; i++)
{
mm[i] = ((i&(i-1)) == 0)?mm[i-1]+1:mm[i-1];
dp[i][0] = b[i];
}
for(int j = 1; j <= mm
; 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]);
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: