JZOJ 5842

Description

给定一个n*m 的 01 矩阵，求包含[l,r]个 1 的子矩形个数。

Input

第一行，两个正整数n，m。
接下来n 行，每行一个长度为 m 的 01 串，表示给定的矩阵。接下来一行，两个自然数 l，r。

Output

第一行，一个整数代表答案。

Data Constraint

做法：由于n很小，我们可以枚举n，统计答案。

 #include <cstdio>
 #include <iostream>
 #include <cstring>
 #define LL long long
 #define rep(i,a,b) for(int i=a;i<=b;i++)
 using namespace std;
 int n,m;
 int a[][],Q[][],f[],L,R;
 LL ans;

 void Init(){
     scanf("%d%d",&n,&m);
     char ch=getchar();
     rep(i,,n){
         rep(j,,m){
             char ch=getchar();
             a[i][j]=ch-'';
             Q[i][j]=Q[i-][j]+Q[i][j-]-Q[i-][j-]+a[i][j];
         }
         char ch=getchar();
     }
     scanf("%d%d",&L,&R);
 }

 void Calc(int up,int down){
     rep(i,,m)    f[i]=Q[down][i]-Q[up-][i];
     int l=,r=;
     rep(i,,m){
         for(;f[i]-f[r]>=L&&r<i;r++);
         for(;f[i]-f[l]>R&&l<i;l++);
         ans+=r-l;
     }
 }

 void Work(){
     ans=;
     rep(i,,n)
         rep(j,i,n)
             Calc(i,j);
     printf("%lld",ans);
 }

 int main(){
     freopen("a.in","r",stdin);
     freopen("a.out","w",stdout);
     Init();
     Work();
 }