子矩阵 单调队列

NKOJ 3681 子矩阵

问题描述

小 A 有一个 N×M 的矩阵，矩阵中 1~N*M 这(N*M)个整数均出现过一次。 现在小 A 在这个矩阵内选择一个子矩阵，其权值等于这个子矩阵中的所有数的最 小值。小 A 想知道，如果他选择的子矩阵的权值为 i(1<=i<=N×M)，那么他选择 的子矩阵可能有多少种？小 A 希望知道所有可能的 i 值对应的结果，但是这些结 果太多了，他算不了，因此他向你求助。

输入格式

第一行，两个整数 N,M。
接下来的 N 行，每行 M 个整数，表示矩阵中的元素。

输出格式

N×M 行，每行一个整数，其中第 i 行的整数表示如果小 A 选择的子矩阵权 值为 i，他选择的子矩阵的种类数。

数据范围

对于 30%的数据，1<=N,M<=50；
对于全部的数据，1<=N,M<=300。

---

首先说说$O(N^2M^2)$的暴力做法：

考虑固定矩阵左上角，如何求出以该点为左上角的所有矩阵的权值？令$f[i][j]$表示以当前左上角，$(i,j)$为右上角构成的矩形的权值，那么有递推关系：

$$f[i][j]=min(f[i-1][j],f[i][j-1],Map[i][j])$$

对于一个固定的左上角，求出所有矩阵的权值的时间复杂度为$O(NM)$。左上角一共讨论$O(NM)$个，所以总时间复杂度$O(N^2M^2)$。

---

正解：

从权值的角度考虑。固定矩阵上下边界后，如何求某一权值的矩形有多少个？那么首先要找到当前权值的点在哪里。显然，当遇到左边第一个权值小于它的矩形就停止，这样就找到了它的左边界。右边界同理。这样的话，以该上下边界的矩阵个数为$(Right[k]-k+1)(k-Left[k]+1)$。

以某点为中心，找左右边界的操作可以通过单调队列（单调栈）用$O(M)$解决。枚举上下边界的时间复杂度是$O(N^2)$，总时间复杂度$O(N^2M)$。当然也可以是$O(NM^2)$。

---

#include<stdio.h>
#define min(x,y) ((x<y)?(x):(y))

int N,M,Min[305],Ans[90005],Map[305][305],Q[305],tail,L[90005],R[90005];

int main()
{
    int i,j,k;

    scanf("%d%d",&N,&M);
    for(i=1;i<=N;i++)
    for(j=1;j<=M;j++)scanf("%d",&Map[i][j]);

    for(i=1;i<=N;i++)
    {
        for(j=1;j<=M;j++)Min[j]=1e9;
        for(j=i;j<=N;j++)
        {
            for(k=1;k<=M;k++)Min[k]=min(Min[k],Map[j][k]);
            tail=0;
            for(k=1;k<=M;k++)
            {
                while(tail&&Min[Q[tail]]>Min[k])
                {
                    R[Q[tail]]=k-1;
                    tail--;
                }
                Q[++tail]=k;
            }
            while(tail)R[Q[tail]]=M,tail--;
            for(k=M;k>=1;k--)
            {
                while(tail&&Min[Q[tail]]>Min[k])
                {
                    L[Q[tail]]=k+1;
                    tail--;
                }
                Q[++tail]=k;
            }
            while(tail)L[Q[tail]]=1,tail--;
            for(k=1;k<=M;k++)Ans[Min[k]]+=(R[k]-k+1)*(k+1-L[k]);
        }
    }

    for(i=1;i<=N*M;i++)printf("%d\n",Ans[i]);
}