POJ 2019 二维RMQ

题意：

给出一个N*N的矩阵，要查询任意B*B子矩阵内的元素最大值和最小值之差。

 

思路：

没神马思路可言。刚刚做完一道RMQ，正好碰到这道题。。果断二维RMQ。

不要听到“二维”就想到二维线段树，应该想到二维树状数组（指变成复杂度）

二维RMQ 和就是在一维的基础上及一层循环

详见代码~

 

#include <cstdio>
#include <cstring>
#include <string>
#include <cstdlib>
#include <iostream>
#include <cmath>

#define N 500

using namespace std;

int pmax[N][N][11],pmin[N][N][10],a[N][N],n,q,d,maxans,minans;

void init_rmq()
{
    for(int i=1;i<=n;i++)
        for(int j=1;j<=n;j++)
            pmax[i][j][0]=pmin[i][j][0]=a[i][j];
    
    for(int i=1;i<=n;i++)
        for(int k=1;(1<<k)<=n;k++)
            for(int j=1;j+(1<<k)-1<=n;j++)
            {
                pmax[i][j][k]=max(pmax[i][j][k-1],pmax[i][j+(1<<(k-1))][k-1]);
                pmin[i][j][k]=min(pmin[i][j][k-1],pmin[i][j+(1<<(k-1))][k-1]);
            }
}

void read()
{
    for(int i=1;i<=n;i++)
        for(int j=1;j<=n;j++)
            scanf("%d",&a[i][j]);
    init_rmq();
}

void askrmq(int a,int b)
{
    int k=(int)(log(double(d))/log(2.0));
    maxans=0; minans=0x3f3f3f3f;
    int l=b,r=b+d-1;
    for(int i=a;i<a+d;i++)
    {
        maxans=max(maxans,max(pmax[i][l][k],pmax[i][r-(1<<k)+1][k]));
        minans=min(minans,min(pmin[i][l][k],pmin[i][r-(1<<k)+1][k]));
    }
}

void go()
{
    for(int i=1,a,b;i<=q;i++)
    {
        scanf("%d%d",&a,&b);
        askrmq(a,b);
        printf("%d\n",maxans-minans);
    }
}

int main()
{
    while(scanf("%d%d%d",&n,&d,&q)!=EOF)
    {
        read();
        go();
    }
    return 0;
}