JZOJ5242【GDOI模拟】矩阵

Description

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

Solution

非常经典的一道题目。
首先我们很容易知道n^3的方法。
那么我们来优化一下。
我们先求出f[i][j]表示从(i,j)向右连续1的数量。
然后把这些从大到小的插入(桶排一下)
然后如果上面或下面已经有了就合并，然后算出贴底的(就是有一边在第j列)的新增矩形方案数，第j+1列插入f[1~n][j+1]的时候会考虑。
然后合并的时候就要考虑跨越中间的分界线且高度不超过h的新增矩形，那么新增矩形就是f[h][size[u]+size[v]]-f[h][size[u]]-f[h][size[v]]
设f[i][j]为在长为i，宽为j的矩形类，贴着j那条边的面积>k的矩形数量。
那么$f[i][j]=f[i][j-1]+\sum_{k=1}^ig[k][i]$
g[i][j]表示底为i，高不大于j，面积>k的矩形数量，可以直接求出来$g[i][j]=j-\lceil{k\over i}\rceil+1$
然后g前缀和一下就好了，改一下枚举i,j的顺序，g数组可以只用一个变量。
合并用并查集维护
复杂度O(n^2logn)
也可以维护每块的最左最右端点，可以省掉log

Code

#include<iostream>
#include<stdio.h>
#include<string.h>
#include<algorithm>
#include<math.h>
#define fo(i,a,b) for(i=a;i<=b;i++)
#define fod(i,a,b) for(i=a;i>=b;i--)
using namespace std;
typedef long long ll;
const int maxn=4007;
int i,j,k,l,t,n,m,x,y,o;
int a[maxn][maxn],f[maxn][maxn],c[maxn];
int b[maxn*maxn][2],fa[maxn*maxn],size[maxn*maxn],p;
bool bz[maxn][maxn];
ll ans,g[maxn][maxn];
int gf(int x){return(!fa[x])?x:gf(fa[x]);}
int get(){
    int x=0;char ch=getchar();
    while(ch<'0'||ch>'1')ch=getchar();
    while(ch>='0'&&ch<='1')x=x*10+ch-'0',ch=getchar();
    return x;
}
void merge(int x,int y,int h){
    int u=gf(x),v=gf(y);
    fa[u]=v;
    ans+=g[h][size[v]+size[u]]-g[h][size[v]]-g[h][size[u]];
    size[v]+=size[u];
}
int de(int x,int y){
    if(x<0||x>n)return 0;
    return (x-1)*m+y;
}
int main(){
    freopen("matrix.in","r",stdin);
    freopen("matrix.out","w",stdout);
    scanf("%d%d%d",&n,&m,&k);
    fo(j,1,m){
        p=0;
        fo(i,1,n){
            p+=max(0,j-(k/i+(k%i>0))+1);
            g[i][j]=g[i][j-1]+p;
        }
    }
    fo(i,1,n)fo(j,1,m){a[i][j]=get();f[i][j]=(!a[i][j])?f[i][j-1]+1:0;c[f[i][j]]++;size[(i-1)*m+j]=1;}
    fo(i,1,m)c[i]+=c[i-1];
    fo(i,1,n)fo(j,1,m)
    b[c[f[i][j]]][0]=i,b[c[f[i][j]]--][1]=j;
    fod(i,n*m,1){
        x=b[i][0],y=b[i][1];if(!f[x][y])break;
        if(f[x][y]>=k)ans+=f[x][y]-k+1;bz[x][y]=1;
        if(bz[x-1][y])merge(de(x-1,y),de(x,y),f[x][y]);
        if(bz[x+1][y])merge(de(x+1,y),de(x,y),f[x][y]);
    }
    printf("%lld\n",ans);
}