POJ-3494(最大全1子矩阵,单调栈)

题目:http://poj.org/problem?id=3494

分析:相当于是最大直方和的二维版,对于最大直方和问题,我们对每个点找到最左边大于等于它的位置和最右边大于等于它的位置,用一个单调栈可以在O(N)空间O(N)时间内完成,对于本题,我们即分别将每一行作为直方图的X轴,所以时间上是O(MN),空间上仍需O(N)辅助


#include 
#define MAX	2005

inline int max(int a, int b){ return a > b ? a : b; }

int M, N, a[MAX][MAX], ans;
int l[MAX], st[MAX], top;

void floor(int f)
{
	l[0] = 1;
	top = 0;
	st[++top] = 0;
	for(int i = 1; i < N; ++i){
		while(top && a[f][st[top]] >= a[f][i]) --top;
		if(top) l[i] = i - st[top];
		else l[i] = i + 1;
		st[++top] = i;
	}

	ans = max(ans, a[f][N-1] * l[N-1]);
	top = 0;
	st[++top] = N-1;
	for(int i = N-2; i > -1; --i){
		while(top && a[f][st[top]] >= a[f][i]) --top;
		//r[i] = N - i - top;
		int r;
		if(top) r = st[top] - i;
		else r = N - i;
		ans = max(ans, a[f][i] * (r + l[i] - 1));
		st[++top] = i;
	}
}

int main()
{
	while(~scanf("%d%d", &M, &N)){
		//input
		for(int i = 0; i < M; ++i)
			for(int j = 0; j < N; ++j)
				scanf("%d", a[i] + j);
		//find out continuous height
		for(int i = 1; i < M; ++i)
			for(int j = 0; j < N; ++j)
				if(a[i][j]) a[i][j] += a[i-1][j];
		//for each floor find out highest
		ans = 0;
		for(int i = 0; i < M; ++i) floor(i);
		printf("%d\n", ans);
	}
	return 0;
}


你可能感兴趣的:(单调队列,单调栈,每天A一道题)