[codeforces 1323B] Count Subrectangles 分而治之+细节较多

Codeforces Round #626 (Div. 2, based on Moscow Open Olympiad in Informatics)   比赛人数5260

[codeforces 1323B]  Count Subrectangles    分而治之+细节较多

总目录详见https://blog.csdn.net/mrcrack/article/details/103564004

在线测评地址https://codeforces.ml/contest/1323/problem/B

Problem	Lang	Verdict	Time	Memory
B - Count Subrectangles	GNU C++11	Accepted	46 ms	800 KB

比赛时，第一直觉，用容斥原理，编着编着，感觉细节较多，果断切换到C题，成功AC后，又切换回该题。

改变算法，不用容斥原理，直来直去处理。

基本想法，是将矩阵分成元素全为1的子矩阵，再对子矩阵进行统计。手工算法如下：

1 1 1 1 0 1 1
1 1 1 1 0 1 1
1 1 1 1 0 1 1
0 0 0 0 0 0 0
1 1 1 1 0 1 1

k=4情况选取
上述矩阵可分为元素全是1的子矩阵

1 1 1 1
1 1 1 1
1 1 1 1
对应k=4有9种可能

1 1
1 1
对应k=4有1种可能

1 1 1 1
对应k=4有1种可能

1 1
对应k=4有0种可能

该题具体编写时，细节较多。

1.int溢出举例

n=40000,m=40000,k=8,矩阵元素全是1,

种类比较简单，

8*1的矩阵数量+4*2的矩阵数量+2*4的矩阵数量+1*8的矩阵数量

(40000-8+1)+(40000-4+1)*(4000-2+1)+(40000-2+1)*(40000-4+1)+(40000-8+1)

上述结果int已经溢出。

2.矩阵分块后，子矩阵可以不考虑出现顺序，进行合并(ans+=solve(r[i].v,c[j].v,k)*r[i].num*c[j].num;)，这样能大大降低，时间复杂度。

3.在处理一串连续序列是，要特别注意边界的处理，编写代码过程中，往往这点是最耗时，也最容易出错的。

4.为了减低代码长度，特别用了函数。

5.代码看起来，套了许多循环，算法的时间度较高，其实读者多举几个极端的例子，就能明白，算法的时间复杂度其实是很低的.

AC代码如下，

他人的代码，有时是较难看懂的，建议读者把握基本思路，进行编写，不明白的地方，再进行对比，看懂的可能性才大。

#include <stdio.h>
#define maxn 40010
#define LL long long
int a[maxn],b[maxn],rn,cn;
struct node{
	int num,v;
}r[maxn],c[maxn];
void count(int n,int *x,node *y,int *c){//分成包含元素全为1的小矩阵块。
	int t,cnt,pre=0,i;
	for(i=1;i<=n;i++){
		scanf("%d",&t);
		if(t){//t==1
			if(pre==t)cnt++;
			else pre=1,cnt=1;
		}else{//t==0
			if(pre==1)x[cnt]++,cnt=0,pre=0;
		}
	}
	if(t)x[cnt]++;//i==n的处理,//包含了n==1的处理
	*c=0;
	for(i=1;i<=n;i++)
		if(x[i])y[++*c].num=x[i],y[*c].v=i;
}
LL solve(int n,int m,int k){
	LL ans=0;
	int i,r,c;
	for(i=m;i>=1;i--)//c=i;
		if(k%i==0){
			c=i,r=k/i;
			if(r<=n)ans+=(n-r+1)*(m-c+1);//判断if(r<=n)不能省
		}
	return ans;
}
int main(){
	int n,m,k,i,j;
	LL ans=0;
	scanf("%d%d%d",&n,&m,&k);
	count(n,a,r,&rn);
	count(m,b,c,&cn);
	for(i=1;i<=rn;i++)
		for(j=1;j<=cn;j++)
			ans+=solve(r[i].v,c[j].v,k)*r[i].num*c[j].num;
	printf("%lld\n",ans);
	return 0;
}