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;
}

 

上一篇:【递归乘法】【100%完美满分算法】【标准解法=快速乘】【反向优化=FFT】


下一篇:LuoguP1635 WJMZBMR打osu! / Easy 题解