牛客多校训练营（第三场） F Planting Trees(单调队列，处理矩阵）

题目大意

给你一个n*n的矩阵，矩阵的每个元素有对应的权值，给一个值m，求一个该矩阵的最大的子矩阵，保证子矩阵中最大权值的元素和最小权值的元素的权值之差不超过m。

思路

发现现在越来越多单调队列和单调栈的问题了，特别是在一个矩阵或者方阵中求一个最大的子矩阵这样类似的问题，一般这种问题我们需要将二维的矩阵降维处理，即将行列分开处理，先处理好一维，然后在去考虑另外一维，而处理一维的问题，单调队列和单调栈又是非常实用的工具。
就像这一题，题目提到了n^3的做法合理，那么我们就可以考虑去枚举一维的区间，然后再处理出答案，单调队列可以帮助我们将区间控制在合法的范围内，于是我们就可以得到答案了。
具体在下面的代码里

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <map>
#include <queue>
#include <cstring>
#include <stack>
#define inf 0x3f3f3f3f
const static int maxn = 510;
using namespace std;
typedef long long ll;
int a[maxn][maxn];
int row_max[maxn];
int row_min[maxn];
int q1[maxn], q2[maxn];
int n, m;


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

void solve()
{
    int ans = 0;
    for(int l=1; l<=n; l++)//枚举左区间
    {
        for(int i=1; i<=n; i++)
        {
            row_max[i] = -inf;
            row_min[i] = inf;
        }
        for(int r=l; r<=n; r++)//枚举右区间
        {
            for(int i=1; i<=n; i++)
            {
                row_max[i] = max(row_max[i], a[i][r]);//更新每一行左右区间内的最值
                row_min[i] = min(row_min[i], a[i][r]);
            }
            int l1, l2, r1, r2;
            l1 = l2 = 1;
            r1 = r2 = 0;
            for(int i=1, j=1; i<=n; i++)//枚举下区间,单调队列处理出在左右区间和上下区间内符合条件位置
            {
                while(l1<=r1 && row_min[q1[r1]] >= row_min[i]) r1--;//对于最小值，在当前枚举区间的上面还比当前枚举区间大的信息已经无用
                                                                    //故可以舍去
                q1[++r1] = i;
                while(l2<=r2 && row_max[q2[r2]] <= row_max[i]) r2--;
                q2[++r2] = i;
                while(j<=i && row_max[q2[l2]] - row_min[q1[l1]] > m)
                {
                    j++;
                    while(l1<=r1 && q1[l1] < j)//保证上端点和单调队列中的信息在合法区间之内
                        l1++;
                    while(l2<=r2 && q2[l2] < j)
                        l2++;
                }
                ans = max(ans, (r-l+1)*(i-j+1));
            }
        }
    }
    printf("%d\n", ans);
}

int main()
{
    int t;
    scanf("%d", &t);
    while(t--)
    {
        input();
        solve();
    }
    return 0;
}
/*
2
2 0
1 2
2 1
3 1
1 3 2
2 3 1
3 2 1
*/