题目大意
给你一个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
*/