2019牛客暑期多校训练营（第三场） F Planting Trees(单调队列)

题目链接：https://ac.nowcoder.com/acm/contest/883/F

 

题目大意：找出最大的矩阵满足其中最大值和最小值的差小于等于m

 

题目思路：首先题目已经非常明显地提醒这是一个n^3的题目，所以就肯定能想到是枚举上下界右界然后O(1)地得到最远的左界，那么问题可以转换为在一个一维的数列中如何O(n)得到一个最长的连续序列使得其中的最大最小值差小于等于m。同时维护两个单调队列，一个维护最大值，一个维护最小值。首先可以发现左界具有单调性，即枚举右界右移后，左界不会向左移动，以维护最大值单调队列为例，如果单调队列尾部的元素比当前值小，那么当左区间在这个元素位置上时，仍以当前值作为最大值，这个元素无效，所以把它踢了，也就是维护最大值的单调队列中维护的是一个递减序列，而维护最小值的正好相反。然后就不断推进左边界，通过单调队列了解左边界到右边界内最大最小值之差，直到满足要为止。

 

以下是代码：

#include<bits/stdc++.h>
using namespace std;
#define inf 0x3f3f3f3f
#define rep(i,a,b) for(ll i=a;i<=b;i++)
#define per(i,a,b) for(ll i=a;i>=b;i--)
#define ll long long
const ll MAXN = 505;
const ll MAXM = 1e6+5;
const ll MOD = 1e9+7;
ll a[MAXN][MAXN],mn[MAXN],mx[MAXN],q1[MAXN],q2[MAXN];
int main(){
    ll t,n,m;
    scanf("%lld",&t);
    while(t--){
        scanf("%lld%lld",&n,&m);
        rep(i,1,n){
            rep(j,1,n)scanf("%lld",&a[i][j]);
        }
        ll ans=0;
        rep(i,1,n){
            rep(j,1,n)mn[j]=mx[j]=a[i][j];
            rep(j,i,n){
                ll l1=1,r1=0,l2=1,r2=0,pos=1;
                rep(k,1,n){
                    mn[k]=min(mn[k],a[j][k]),mx[k]=max(mx[k],a[j][k]);
                    while(l1<=r1&&mn[q1[r1]]>=mn[k])r1--;
                    while(l2<=r2&&mx[q2[r2]]<=mx[k])r2--;
                    q1[++r1]=k,q2[++r2]=k;
                    while(pos<=k&&mx[q2[l2]]-mn[q1[l1]]>m){
                        pos++;
                        if(l1<=r1&&q1[l1]<pos)l1++;
                        if(l2<=r2&&q2[l2]<pos)l2++;
                    }
                    ans=max(ans,(j-i+1)*(k-pos+1));
                }
            }
        }
        printf("%lld\n",ans);
    }
    return 0;
}