有这样一种问题:有一个数列a1,a2,.....,an,我们需要查询某一特定区间上的最小值,如何实现?这类问题被称为RMQ问题(range minimum query),其最经典的解决方案是Tarjian的Sparse-Table算法。这种算法用NlogN的代价初始化,但查询只需要常数时间。
通常来说,log的出现总是和2的幂分不开,这里也采用这种思想(比如多重背包的物品拆分优化)我们求出所有以第i个元素开头,长2的j次方个元素的区间上的最小值,可以用数组d[i][j]来记录,数组元素个数为O(NlogN)。注意这里的递推思想:d(i,j)=min{d(i,j-1),d(i+pow(2,j-1),j-1)}实际上就是把原区间分成了两个部分。初始化的时候要注意一下递推的顺序,首先d[i][0]全都等于本身,之后我们先从j=1开始推出所有的d[i][1]。否则i+pow项会来不及算就参与递推了。之后反复增加j继续这个过程。
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)<n;i++)
d[i][j]=min(d[i][j-1],d[i+(1<<j)][j-1]);
}
查询的时候,我们先找出最大的k,使得pow(2,k)<=R-L+1。则d[L][K],d[R-pow(2,k)+1][R]两个区间(有重合)覆盖L-R,这两个区间中的最小值即为所求。
int RMQ(int L,int R)
{
int k=0;
while(1<<(k+1)<=R-L+1)k++;//当k+1未超出时k还可以加一
return min(d[L][k],d[R-(1<<k)+1][k]);
}
注:代码部分参考了刘汝佳的《算法竞赛入门经典(训练指南)》