B1047 理想的正方形 RMQ

大家吸取我的教训，想好再写。我码了好长时间，最后发现算法处理的是另一个问题，我处理的是正方形情况的，才能用我优化之后的记忆化搜索，然而矩形就凉了。

先看一下题吧：

Description

　　有一个a*b的整数组成的矩阵，现请你从中找出一个n*n的正方形区域，使得该区域所有数中的最大值和最小值
的差最小。
Input

　　第一行为3个整数，分别表示a,b,n的值第二行至第a+1行每行为b个非负整数，表示矩阵中相应位置上的数。每
行相邻两数之间用一空格分隔。
%的数据2<=a,b<=,n<=a,n<=b,n<=
Output

　　仅一个整数，为a*b矩阵中所有“n*n正方形区域中的最大整数和最小整数的差值”的最小值。
Sample Input

Sample Output

HINT

Source

我的反面教材代码：

#include<iostream>
#include<cstdio>
#include<cmath>
#include<ctime>
#include<queue>
#include<algorithm>
#include<cstring>
using namespace std;
#define duke(i,a,n) for(int i = a;i <= n;i++)
#define lv(i,a,n) for(int i = a;i >= n;i--)
#define clean(a) memset(a,0,sizeof(a))
const int INF =  << ;
typedef long long ll;
typedef double db;
template <class T>
void read(T &x)
{
    char c;
    bool op = ;
    while(c = getchar(), c < '' || c > '')
        if(c == '-') op = ;
    x = c - '';
    while(c = getchar(), c >= '' && c <= '')
        x = x *  + c - '';
    if(op) x = -x;
}
template <class T>
void write(T x)
{
    if(x < ) putchar('-'), x = -x;
    if(x >= ) write(x / );
    putchar('' + x % );
}
int a,b,n;
int p[][];
int f[][][][];
int l[];
void dfs(int o,int x1,int y1,int x2,int y2)
{
    if(f[x1][y1][o][] != -)
        return;
    if(x1 == x2 && y1 == y2)
    {
        f[x1][y1][o][] = p[x1][y1];
        f[x1][y1][o][] = p[x1][y1];
        return;
    }
    int minn = INF;
    int maxn = ;
    dfs(o + ,x1,y1,x1 + l[o],y1 + l[o]);
    dfs(o + ,a - l[o],y1,a,y1 + l[o]);
    dfs(o + ,x1,b - l[o],x1 + l[o],b);
    dfs(o + ,a - l[o],b - l[o],a,b);
//    printf("%d %d %d %d\n",a - l[o],b - l[o],a,b);
    maxn = max(maxn,f[x1][y1][o + ][]);
    minn = min(minn,f[x1][y1][o + ][]);
    maxn = max(maxn,f[a - l[o]][y1][o + ][]);
    minn = min(minn,f[a - l[o]][y1][o + ][]);
    maxn = max(maxn,f[x1][b - l[o]][o + ][]);
    minn = min(minn,f[x1][b - l[]][o + ][]);
    maxn = max(maxn,f[a - l[o]][b - l[o]][o + ][]);
    minn = min(minn,f[a - l[o]][b - l[o]][o + ][]);
    f[x1][y1][o][] = maxn;
    f[x1][y1][o][] = minn;
    return;
}
int main()
{
    memset(f,-,sizeof(f));
    read(a);read(b);read(n);
    duke(i,,a)
    {
        duke(j,,b)
            read(p[i][j]);
    }
    int k = n,len = ;
    while(k > )
    {
        k = ceil(k / );
        l[++len] = k;
//        cout<<k<<endl;
    }
    duke(i,,a - n)
        duke(j,,b - n)
            dfs(,i,j,i + n,j + n);
    printf("%d %d\n",f[][][][],f[][][][]);
    return ;
}
/*
5 4 2
1 2 5 6
0 17 16 0
16 17 2 1
2 10 2 1
1 2 2 2
*/

然后就凉了，懒得写了，用二维的RMQ搞一下，抄了个比较清晰的代码。。。

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

const int INF = ;
const int maxm =  + ;
const int maxn =  + ;
const int maxlog = ;

int a, b, n;
int logn;
int grid[maxm][maxm];
int maxv[maxm][maxm], minv[maxm][maxm];

int query (int x, int y)
{
    int _max = , _min = ;
    _max = max(maxv[x][y], max(maxv[x+n-(<<logn)][y+n-(<<logn)], max(maxv[x+n-(<<logn)][y], maxv[x][y+n-(<<logn)])));
    _min = min(minv[x][y], min(minv[x+n-(<<logn)][y+n-(<<logn)], min(minv[x+n-(<<logn)][y], minv[x][y+n-(<<logn)])));
    return _max - _min;
}

int main ()
{
//    freopen("in.txt", "r", stdin);
    cin >> a >> b >> n;
    for (int i = ; i < a; i++)
        for (int j = ; j < b; j++)
        {
            scanf("%d", &grid[i][j]);
            maxv[i][j] = minv[i][j] = grid[i][j];
        }

    for (logn = ; ((<<(logn+)) <= n); logn++);
    for (int k = ; k < logn; k++)
        for (int i = ; i+(<<k) < a; i++)
            for (int j = ; j+(<<k) < b; j++)
            {
                maxv[i][j] = max(maxv[i][j], max(maxv[i+(<<k)][j+(<<k)], max(maxv[i+(<<k)][j], maxv[i][j+(<<k)])));
                minv[i][j] = min(minv[i][j], min(minv[i+(<<k)][j+(<<k)], min(minv[i+(<<k)][j], minv[i][j+(<<k)])));
            }

    int ans = INF;
    for (int i = ; i <= a-n; i++)
        for (int j = ; j <= b-n; j++)
            ans = min(ans, query(i, j));

    cout << ans;
    return ;
}

唉，后悔啊。。。