矩阵习题

2020牛客第九场 J The Escape Plan of Groundhog （枚举 + 前缀和）

链接：https://ac.nowcoder.com/acm/contest/5674/J

题意：给定一个 $n\times m$ 由 01 组成的矩阵，问有多少个子矩阵满足，周围一圈全部为 1 ，内部 0 和 1 的个数之差不超过 1 。输出矩阵的数量。$（1\le n,m\le 500）$

思路：对于矩阵的题，通常枚举上界和下界，最后再枚举列

• 枚举列时，固定 l ，然后 r 不断往右扩展，当遇到第 r 列全为 1 时，记录由（i ，1）（j ，r）构成矩阵的前缀和出现次数，放在 cnt 数组中。
• 这样在下一次遇到全为 1 时的列时，就可以先统计答案，累加 由（i ，1）（j - 1 ，r - 1）构成矩阵的前缀和且和它相差 1 的前缀和
• 注意要及时清空 cnt 数组

#include <bits/stdc++.h>
#define ll long long
using namespace std;
const int N=500+10;
int n,m,a[N][N],M=N*N;
int b[N][N],pref[N][N],cnt[N*N*2];
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1; i<=n; ++i)
        for(int j=1; j<=m; ++j)
        {
            scanf("%d",&a[i][j]);
            if(a[i][j]==0) a[i][j]=-1;
            b[i][j]=b[i-1][j]+a[i][j];
            pref[i][j]=pref[i-1][j]+pref[i][j-1]-pref[i-1][j-1]+a[i][j];
        }
    int ans=0;
    for(int i=1; i<=n; ++i)
    {
        for(int j=i+1; j<=n; ++j)
        {
            int l=1;
            for(int r=1; r<=m; ++r)
            {
                if(a[i][r]!=1||a[j][r]!=1)
                {
                    for(int k=l; k<=r; ++k)
                        if(b[j][k]-b[i-1][k]==j-i+1) cnt[pref[j-1][k]-pref[i][k]+M]--;
                    l=r+1;
                    continue;
                }
                if(b[j][r]-b[i-1][r]==j-i+1)
                {
                    int res=pref[j-1][r-1]-pref[i][r-1]+M;
                    ans+=cnt[res-1]+cnt[res]+cnt[res+1];
                    cnt[pref[j-1][r]-pref[i][r]+M]++;
                }
            }
            for(int r=l; r<=m; ++r)
                if(b[j][r]-b[i-1][r]==j-i+1) cnt[pref[j-1][r]-pref[i][r]+M]--;
        }
    }
    printf("%d\n",ans);
    return 0;
}

2019牛客第三场 F Planting Trees （枚举 + 单调队列）

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

题意：给定一个 $n\times n$ 的矩阵，找出最大的子矩阵要求，子矩阵中的最大值 - 最小值不超过 k 。输出这个子矩阵的大小。$（1\le n\le 500）$

思路：维护单调递增和单调递减两个队列。

• 对于矩阵的题，枚举上边界、下边界，然后枚举右边界，用 l 来控制左边界。当最大值 - 最小值 超过 k 时，要确定具体缩哪个队列是很难做到的。但是可以用 l 来收缩队列，让在 l 左边的 h1、h2 不断缩减就好了。然后继续循环判断是否满足条件。

#include <bits/stdc++.h>
#define ll long long
using namespace std;
const int maxn=500+10;

int t,n,m;
int a[maxn][maxn];
int mx[maxn],mn[maxn];
int q1[maxn],q2[maxn],h1,t1,h2,t2;

int main()
{
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d%d",&n,&m);
        for(int i=1; i<=n; ++i)
            for(int j=1; j<=n; ++j)
                scanf("%d",&a[i][j]);
        int ans=0;
        for(int i=1; i<=n; ++i)
        {
            for(int j=i; j<=n; ++j)
            {
                if(i==j)
                    for(int k=1; k<=n; ++k) mx[k]=mn[k]=a[i][k];
                else
                {
                    for(int k=1; k<=n; ++k)
                    {
                        mx[k]=max(mx[k],a[j][k]);
                        mn[k]=min(mn[k],a[j][k]);
                    }
                }
                int l=1;
                h1=h2=1,t1=t2=0;
                for(int r=1; r<=n; ++r)
                {
                    while(h1<=t1&&mn[r]<=mn[q1[t1]]) t1--;
                    q1[++t1]=r;

                    while(h2<=t2&&mx[r]>=mx[q2[t2]]) t2--;
                    q2[++t2]=r;

                    while(h1<=t1&&h2<=t2&&mx[q2[h2]]-mn[q1[h1]]>m)
                    {
                        l++;
                        while(h1<=t1&&q1[h1]<l) h1++;
                        while(h2<=t2&&q2[h2]<l) h2++;
                    }
                    ans=max(ans,(r-l+1)*(j-i+1));
                }
            }
        }
        printf("%d\n",ans);
    }
    return 0;
}