poj 3368 RMQ 模版应用

#include<iostream>
#include<cmath>
#include<cstdlib>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
#define M 111111
#define MAXN 500
#define MAXM 500
int dp[M][30];
int dp2[MAXN][MAXM][10][10];
int l[M],r[M],a[M];
/*
*一维RMQ ST算法
*构造RMQ数组 makermq(int n,int b[]) O(nlog(n))的算法复杂度
*dp[i]j] 表示从i到i+2^j -1中最大的一个值
*dp[i][j]=max{dp[i][j-1],dp[i+2^(j-1)][j-1]}
*查询RMQ rmq(int s,int v)
*将s ->v 分成两个2^k的区间
*即 k=(int)log2(s-v+1)
*查询结果应该为 max(dp[s][k],dp[v-2^k+1][k])
*/
int rmq(int s,int v)
{
if(l[v]<=s||r[s]>=v)return v-s+1;

//int ans=r[v-(1<<k)]-l[v-(1<<k)]+1;
int ans=max(r[s]-s+1,v-l[v]+1);
s=r[s]+1,v=l[v]-1;// cout<<"OK "<<ans<<endl;
if(s>v)return ans;
int k=(int)(log((v-s+1)*1.0)/log(2.0));
return max(ans,max(dp[s][k],dp[v-(1<<k)+1][k]));
}
void makermq(int n,int b[])
{
int i,j;
for(i=1;i<=n;i++)
dp[i][0]=i-b[i]+1;
for(j=1;(1<<j)<=n;j++)
for(i=1;i+(1<<j)-1<=n;i++)
dp[i][j]=max(dp[i][j-1],dp[i+(1<<(j-1))][j-1]);
}

/*
*二维RMQ ST算法
*构造RMQ数组 makermq(int n,int m,int b[][]) O(n*m*log(n)*log(m))算法复杂度
*dp2[row][col][i][j] 表示 行从row ->row +2^i-1 列从col ->col +2^j-1 二维区间里最大值
*dp2[row][col][i][j] = 下行
*max{dp2[row][col][i][j-1],dp2[row][col][i-1][j],dp2[row][col+2^(j-1)][i][j-1],dp2[row+2^(i-1)][col][i-1][j]}
*查询RMQ rmq(int sx,int ex,int sy,int ey)
*同一维的将sx->ex 分为两个2^kx区间 将 sy->ey分为两个2^ky的区间
*kx=(int)log2(ex-sx+1) ky=(int)log2(ey-sy+1)
*查询结果为
*max{dp2[sx][sy][kx][ky],dp2[sx][ey-2^ky+1][kx][ky],dp2[ex-2^kx+1][sy][kx][ky],dp2[ex-2^kx+1][ey-2^ky+1][kx][ky]}
*/

void makermq(int n,int m,int b[][MAXM])
{
int row,col,i,j;
for(row=1;row<=n;row++)
for(col=1;col<=m;col++)
dp2[row][col][0][0]=b[row][col];
for(i=0;(1<<i)<=n;i++)
for(j=0;(1<<j)<=m;j++)
{
if(i==0&&j==0) continue;
for(row=1;row+(1<<i)-1<=n;row++)
for(col=1;col+(1<<j)-1<=m;col++)
{
if(i==0)
dp2[row][col][i][j]=max(dp2[row][col][i][j-1],dp2[row][col+(1<<(j-1))][i][j-1]);
else
dp2[row][col][i][j]=max(dp2[row][col][i-1][j],dp2[row+(1<<(i-1))][col][i-1][j]);
}
}
}
int rmq(int sx,int ex,int sy,int ey)
{
int kx=(int)(log((ex-sx+1)*1.0)/log(2.0)),ky=(int)(log((ey-sy+1)*1.0)/log(2.0));
return max(max(dp2[sx][sy][kx][ky],dp2[sx][ey-(1<<ky)+1][kx][ky]),max(dp2[ex-(1<<kx)+1][sy][kx][ky],dp2[ex-(1<<kx)+1][ey-(1<<ky)+1][kx][ky]));
}
int main()
{
int u,v,n,q;
// freopen("//media/学习/ACM/input.txt","r",stdin);
while(scanf("%d",&n),n)
{
scanf("%d",&q);
int i,j;
for(i=1;i<=n;i++)
scanf("%d",&a[i]),l[i]=r[i]=i;
for(i=2;i<=n;i++)
{
if(a[i]==a[i-1])l[i]=l[i-1];
}
for(i=n-1;i>=1;i--)
{
if(a[i+1]==a[i])r[i]=r[i+1];
}
makermq(n,l);
while(q--)
{
scanf("%d%d",&u,&v);
printf("%d\n",rmq(u,v));
}
}
return 0;
}