All-one Matrices 2019牛客多校第8场A题

传送门：https://ac.nowcoder.com/acm/contest/888/A

这场我咕咕咕了，请假回了家，我远程跟队友口胡，队友还没做出来，然后回来后补题竟然还一直WA。。。难顶，hkt给了我几组数据终于调过了。

之前求最大全0子矩阵的经典问题我的单调栈写得是有问题的，对于当前高度h[i][j]相等的情况，如果弹出，那么右区间就不对，如果不弹出，那么左区间就不对，但最右边的那个位置左区间和右区间都是对的，所以可以算出最大全0子矩阵的面积，但是左右区间是不一定对的，所以需要两个方向做两遍单调栈，保证每个位置的左右区间都是对的。

那么，我们考虑这题怎么做，我们可以预处理出来每个位置向上的1的个数h[i][j]，和每个位置能统计的左右区间l[i][j]和r[i][j]

然后我们对于每一行的相同高度，统计出nxt[i][j]就是这个位置他在第i行下一个相同h[i][j]在哪。

那么统计矩形的时候，我们以每行，每个矩形的高度的最右边的h[i][j]为标准统计（这句话我也不太会表述。。。看后面的判断应该能懂）

那么如果 nxt[i][j]<=r[i][j] ，说明第i行高度为h[i][j]的在当前j的右边还有，而且j和nxt[j]属于同一个矩形，那么久不在j这个位置计数+1

如果h[i+1][j]=h[i][j]+1，且l[i+1][j]<=l[i][j] &&r[i+1][j]>=r[i][j]，则说明这个从l[i]到r[i]的矩形还可以向下延伸，所以就不计数

(我一开始没考虑要r[i+1][j]>=r[i][j]wa了)程序下面给出了几组我当时错的数据

#include<bits/stdc++.h>
#define maxl 3010
using namespace std;

int n,m,top,ans;
int a[maxl][maxl],h[maxl][maxl],nxt[maxl][maxl],last[maxl];
int l[maxl][maxl],r[maxl][maxl],s[maxl];
char ch[maxl];

inline void prework()
{
	for(int i=1;i<=n;i++)
	{		
		scanf("%s",ch+1);
		top=0;s[0]=0; 
		for(int j=1;j<=m;j++)
		{
			a[i][j]=ch[j]-'0';
			if(a[i][j])
				h[i][j]=h[i-1][j]+1;
			else
				h[i][j]=0;
			while(top>0 && h[i][j]<=h[i][s[top]])
				top--;
			s[++top]=j;l[i][j]=s[top-1]+1;
		}
		s[0]=m+1;top=0;
		for(int j=m;j>=1;j--)
		{
			while(top>0 && h[i][j]<=h[i][s[top]])
				top--;
			s[++top]=j;r[i][j]=s[top-1]-1;
		}
		for(int i=0;i<=n;i++)
			last[i]=0;
		for(int j=m;j>=1;j--)
		{
			nxt[i][j]=last[h[i][j]];
			last[h[i][j]]=j;
		}
	}
}

inline void mainwork()
{
	ans=0;
	for(int i=1;i<=n;i++)
		for(int j=1;j<=m;j++)
		{
			if(h[i][j]==0) continue;
			if(j<m && nxt[i][j]<=r[i][j] && nxt[i][j]>0)
				continue;
			if(i<n && l[i+1][j]<=l[i][j] && r[i+1][j]>=r[i][j] && h[i+1][j]==h[i][j]+1)
				continue;
			ans++;
		}
}

inline void print()
{
	printf("%d\n",ans);
}

int main()
{
	while(~scanf("%d%d",&n,&m))
	{
		prework();
		mainwork();
		print();
	}
	return 0;
}
/*
4 6
110011
111111
101111
001110
7 10
1011100110
0100111010
1011101101
1111011011
0110111010
1010111011
1111111011
9 10
0111110010
0111111001
1011101000
0111110110
1110010010
1101101000
0000100011
1100111010
1010011111
*/