POJ-2019(Cornfields)

题意：给定一个 N x N的矩阵，有 K 个查询，每次查询给出一个坐标 P(x,y)，求以 P 点为左上角，大小为 B x B 的子矩阵的最大值与最小值的差。

 

分析：二维RMQ裸题。

 

代码：

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;

const int N = 250+5;

int dmax[N][N][8][8];
int dmin[N][N][8][8];
int n,b,q;

void init()
{
    for(int i=0;(1<<i)<=n;i++)
    for(int j=0;(1<<j)<=n;j++)
    {
        if(i==0&&j==0) continue;
        for(int x=1;x+(1<<i)-1<=n;x++)
        for(int y=1;y+(1<<j)-1<=n;y++)
        {
            if(i==0)
            {
                 dmax[x][y][i][j]=max(dmax[x][y][i][j-1],dmax[x][y+(1<<(j-1))][i][j-1]);
                 dmin[x][y][i][j]=min(dmin[x][y][i][j-1],dmin[x][y+(1<<(j-1))][i][j-1]);
            }
            else if(j==0)
            {
                 dmax[x][y][i][j]=max(dmax[x][y][i-1][j],dmax[x+(1<<(i-1))][y][i-1][j]);
                 dmin[x][y][i][j]=min(dmin[x][y][i-1][j],dmin[x+(1<<(i-1))][y][i-1][j]);
            }
            else
            {
                 dmax[x][y][i][j]=max(dmax[x][y][i][j-1],dmax[x][y+(1<<(j-1))][i][j-1]);
                 dmin[x][y][i][j]=min(dmin[x][y][i][j-1],dmin[x][y+(1<<(j-1))][i][j-1]);
            }
        }
    }
}

int query(int x0,int y0,int x1,int y1)
{
    int cntx=0,cnty=0;
    while(x0+(1<<cntx)<=x1) cntx++;
    while(y0+(1<<cnty)<=y1) cnty++;
    cntx--,cnty--;

    int m1=dmax[x0][y0][cntx][cnty];
    int m2=dmax[x1-(1<<cntx)+1][y0][cntx][cnty];
    int m3=dmax[x0][y1-(1<<cnty)+1][cntx][cnty];
    int m4=dmax[x1-(1<<cntx)+1][y1-(1<<cnty)+1][cntx][cnty];
    int Max=max(max(m1,m2),max(m3,m4));

    m1=dmin[x0][y0][cntx][cnty];
    m2=dmin[x1-(1<<cntx)+1][y0][cntx][cnty];
    m3=dmin[x0][y1-(1<<cnty)+1][cntx][cnty];
    m4=dmin[x1-(1<<cntx)+1][y1-(1<<cnty)+1][cntx][cnty];
    int Min=min(min(m1,m2),min(m3,m4));

    return Max-Min;
}

int main()
{
    scanf("%d%d%d",&n,&b,&q);
    for(int i=1;i<=n;i++)
    for(int j=1;j<=n;j++)
    {
        int k;
        scanf("%d",&k);
        dmax[i][j][0][0]=dmin[i][j][0][0]=k;
    }
    init();
    while(q--)
    {
        int x,y;
        scanf("%d%d",&x,&y);
        printf("%d\n",query(x,y,x+b-1,y+b-1));
    }
}