P3400【仓鼠窝 】

思路清奇，代码简洁的好题

问题大体分两部：

• 记录子矩阵个数，统计每一个点作为右下角时可以得到多少矩形，加起来就是答案
• 剪掉墙挡住的地方

考虑从右下角开始，如果有0已经挡住了矩阵，那么更靠左、上的0都卵用没有->单调栈

#include<bits/stdc++.h>
using namespace std;
int n,m,a[3005][3005];
int up[3005],s[3005],top;
long long sum[3005],ans=0;
int main()
{
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++)
	{
		for(int j=1;j<=m;j++)
		{
			scanf("%d",&a[i][j]);
		}
	}
	for(int i=1;i<=n;i++)
	{
		top=0;//维护最近的0
		for(int j=1;j<=m;j++)
		{
			if(!a[i][j]) up[j]=i;//墙
			while(top&&up[s[top]]<up[j]) top--; //远的0不要
			s[++top]=j;//高度
			sum[top]=sum[top-1]+(i-up[s[top]])*(s[top]-s[top-1]);//左边的答案继承+新形成的矩形方案（新的最大矩形长 ×宽）
			ans+=sum[top];
		}
	}
	printf("%lld",ans);
	return 0;
}