JZOJ5242【GDOI2018模拟8.8】矩阵

Description

给出一个n*m 的01 矩阵，求其中有多少个连续子矩阵，使得其面积大于等于k，且其中不包含1。

Input

第一行为用空格隔开的三个整数n, m, k。分别表示矩阵的行数，列数和子矩阵的最小面积。
接下来的n 行每行为用空格隔开的m 个整数，为题目中给出的矩阵。

Output

输出一行一个整数，为满足条件的子矩阵个数。

Sample Input

2 4 3
1 0 0 0
1 1 0 0

Sample Output

2

Data Constraint

对20%的数据，满足1 <= n, m <= 100;
对40%的数据，满足1 <= n, m <= 500;
对60%的数据，满足1 <= n, m <= 2000;
对100%的数据，满足1 <= n, m <= 4,000，1 <= k <= n * m;

Hint

设(a, b, c, d)表示左上角为(a, b)，右下角为(c, d)的子矩阵。
面积为1 的子矩阵:(1, 2, 1, 2), (1, 3, 1, 3), (1, 4, 1, 4), (2, 3, 2, 3), (2, 4, 2, 4);
面积为2 的子矩阵:(1, 2, 1, 3), (1, 3, 1, 4), (2, 3, 2, 4), (1, 3, 2, 3), (1, 4, 2, 4);
面积为3 的子矩阵:(1, 2, 1, 4);
面积为4 的子矩阵:(1, 3, 2, 4)。
其中面积大于等于3 的有两个。

其实这题一点都不难，只是要想细一点。。
预处理出每个0能往右移动的最多位置。
对于每一列上的点，用单调栈处理，预处理个tot[i][j]表示前缀和，就可以方便的求出答案了。

include

include

include

define fo(i,a,b) for(int i=a;i<=b;i++)

define fd(i,a,b) for(int i=a;i>=b;i–)

using namespace std;
const int N=4e3+5;
const int M=1e5+5;
typedef long long ll;
int mp[N][N];
int n,m,h[N][N],a[M],b[M];
ll tot[N][N],right[N][N],ans;
bool vis[N][N];
int sta[M],top;
int s;
int main()
{
scanf(“%d%d%d”,&n,&m,&s);
fo(i,1,n)
{
fo(j,1,m)
{
scanf(“%d”,&mp[i][j]);
}
}
fo(i,1,n)
fd(j,m,1)
if (mp[i][j]!=1)
right[i][j]=right[i][j+1]+1;

fo(i,1,n)
fo(j,1,m)
{
    if (i*j>=s)
    tot[i][j]=1;else tot[i][j]=0;
    tot[i][j]+=tot[i-1][j]+tot[i][j-1]-tot[i-1][j-1];
}
ll ans=0;
fo(j,1,m)
{
    top=0;
    fo(i,1,n)
    {
        if (mp[i][j])top=0;
        else
        {
            int tmp=1,x=i;
            while (top>0&&sta[top]>=right[i][j])x=a[top--];
            sta[++top]=right[i][j];
            a[top]=x,b[top]=i;
            fo(k,1,top)
            ans+=tot[i-a[k]+1][sta[k]]-tot[i-b[k]][sta[k]];
            //printf("%d %d %d\n",i-a[k]+1,i-b[k],sta[k]);
            //printf("\n");
        }
    }
}
/*fo(i,1,n)
{
    fo(j,1,m)printf("%lld ",tot[i][j]);
    printf("\n");
}*/
printf("%lld\n",ans);

}