AcWing每日一题 3617. 子矩形计数（后缀和）

原题链接

给定一个长度为 n 的数组 a 和一个长度为 m 的数组 b。

两个数组均只包含 0 和 1。

利用两个给定数组生成一个 n×m 的矩阵 c，其中 cij=ai×bj。

显然，矩阵 c 中也只包含 0 和 1。

请问，矩阵 c 中有多少个大小（面积）恰好为 k 且只包含 1 的子矩形？

子矩形是指矩阵中连续若干行和连续若干列的交集。

例如，考虑四个整数 x1,x2,y1,y2(1≤x1≤x2≤n,1≤y1≤y2≤m)，子矩形 c[x1…x2][y1…y2] 即为行
x1,x1+1,…,x2 和列 y1,y1+1,…,y2 的一个交集。

一个子矩形的大小（面积）等于它包含的数字个数。

数据范围 1≤n,m≤40000, 1≤k≤n×m, 0≤ai,bi≤1

输入格式
第一行包含三个整数 n,m,k。

第二行包含 n 个整数 a1,…,an，表示数组 a 中的元素。

第三行包含 m 个整数 b1,…,bm，表示数组 b 中的元素。

输出格式
输出满足条件的子矩形的总数量。

---

矩阵中每个元素a[i][j]=r[i]*c[j]
所以对于面积为k的矩形，我们只需要处理出，行有多少个长度为1、2、3…
列有多少个长度为1、2、3…
这样面积就是直接长乘宽就好，再枚举因子即可。

那我们该怎么处理呢
如果有1111 ，4个连续的1
他包含有4个长度1
3个长度2
2个长度3
1个长度4
也就是4321
我们就会发现，后缀和！

AC代码：

#include<bits/stdc++.h>
#define ll long long 
using namespace std;
const int maxn=40010;
int r[maxn],c[maxn];
int sr[maxn],sc[maxn];
int n,m;
ll k;
int main(){
	scanf("%d %d %lld",&n,&m,&k);
	for(int i=1;i<=n;++i)scanf("%d",&r[i]);
	for(int i=1;i<=m;++i)scanf("%d",&c[i]);
	int sumr=0,sumc=0;
	for(int i=1;i<=n;++i){
		if(r[i]==1){
			sumr++;
			sr[sumr]++;
		}
		else sumr=0;
	}
	for(int i=1;i<=m;++i){
		if(c[i]==1){
			sumc++;
			sc[sumc]++;
		}
		else sumc=0;
	}
	for(int i=n-1;i>=1;--i)sr[i]+=sr[i+1];
	for(int j=m-1;j>=1;--j)sc[j]+=sc[j+1];
	ll sum=0;
	for(int i=1;i<=40000&&i<=k;++i){
		if(k%i==0){
		//如果是k的因子
			if(i<=n&&k/i<=m)sum+=sr[i]*sc[k/i];
			//注意要再判断是否在范围之内，防止出现越界
		}
	}
	printf("%lld",sum);
	return 0;
}