[前缀和][枚举] Jzoj P5842 a

Description

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

 

Input

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

Output

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

 

Sample Input

见下发文件

Sample Output

见下发文件
 

 

Data Constraint

 

 

题解

• 注意到n很小，那么可以n^2枚举矩阵的上下边界
• 然后两个指针,x和y,记录到刚好小于l，和刚好大于r的
• 算一波方案数就好了

代码

#include <cstdio>
using namespace std;
char s[50010];
long long n,m,i,j,p[50010],a[40][50010],sum[40][50010],q[50010],ans,k,l,r;
int main()
{
    freopen("a.in","r",stdin);
    freopen("a.out","w",stdout);
    scanf("%lld%lld",&n,&m);
    for (int i=1;i<=n;i++)
    {
        scanf("%s\n",s+1);
        for (int j=1;j<=m;j++)
        {
            a[i][j]=s[j]-'0';
            sum[i][j]=sum[i-1][j]+a[i][j];
        }
    }
    scanf("%lld%lld\n",&l,&r);
    for (int i=1;i<=n;i++)
        for (int j=i;j<=n;j++)
        {
            long long x=0,y=0;
            for (int k=1;k<=m;k++)
            {
                p[k]=sum[j][k]-sum[i-1][k];
                q[k]=q[k-1]+p[k];
            }
            for (int k=1;k<=m;k++)
            {
                while (q[k]-q[x]>=l&&x<k) x++;
                while (q[k]-q[y]>r&&y<k) y++;
                ans+=x-y;
            } 
        }
    printf("%lld\n",ans);
    return 0;
}