1. 程式人生 > >POJ2559 最大矩形面積

POJ2559 最大矩形面積

給定從左到右多個矩形,已知這此矩形的寬度都為1,長度不完全相等。這些矩形相連排成一排,求在這些矩形包括的範圍內能得到的面積最大的矩形,打印出該面積。所求矩形可以橫跨多個矩形,但不能超出原有矩形所確定的範圍。

建立一個單調遞增棧,所有元素各進棧和出棧一次即可。每個元素出棧的時候更新最大的矩形面積。

設棧內的元素為一個二元組(x, y),x表示矩形的高度,y表示矩形的寬度。

若原始矩形高度分別為2,1,4,5,1,3,3

高度為2的元素進棧,當前棧為(2,1)

高度為1的元素準備進棧,但必須從棧頂開始刪除高度大於或等於1的矩形,因為2已經不可能延續到當前矩形。刪除(2,1)這個元素之後,更新最大矩形面積為2*1=2,然後把它的寬度1累加到當前高度為1的準備進棧的矩形,然後進棧,當前棧為(1,2)

高度為4的元素進棧,當前棧為(1,2) (4,1)

高度為5的元素進棧,當前棧為(1,2) (4,1) (5,1)

高度為1的元素準備進棧,刪除(5,1)這個元素,更新最大矩形面積為5*1=5,把1累加到下一個元素,得到(4,2),刪除(4,2),更新最大矩形面積為4*2=8,把2累加到下一個元素,得到(1,4),1*4=4<8,不必更新,刪除(1,4),把4累加到當前準備進棧的元素然後進棧,當前棧為(1,5)

高度為3的元素進棧,當前棧為(1,5) (3,1)

高度為3的元素準備進棧,刪除(3,1),不必更新,把1累加到當前準備進棧的元素然後進棧,當前棧為(1,5) (3,2)

把餘下的元素逐個出棧,(3,2)出棧,不必更新,把2累加到下一個元素,當前棧為(1,7),(1,7)出棧,不必更新。棧空,結束。

最後的答案就是8。

#include <iostream>
#include <cstdio>
using namespace std;

const int N = 100005;

struct Elem
{
	int height;
	int count;
};

Elem stack[N];
int top;

int main()
{
	int height, n;
	long long ans, tot, tmp;
	while (scanf("%d", &n) != EOF && n)
	{
		top = 0;
		ans = 0;
		for (int i = 0; i < n; ++i)
		{
			scanf("%d", &height);
			tmp = 0;
			while (top > 0 && stack[top - 1].height >= height)
			{
				tot = stack[top - 1].height * (stack[top - 1].count + tmp);
				if (tot > ans) ans = tot;
				tmp += stack[top - 1].count;
				--top;
			}
			stack[top].height = height;
			stack[top].count = 1 + tmp;
			++top;
		}
		tmp = 0;
		while (top > 0)
		{
			tot = stack[top - 1].height * (stack[top - 1].count + tmp);
			if (tot > ans) ans = tot;
			tmp += stack[top - 1].count;
			--top;
		}
		printf("%lld\n", ans);
	}
	return 0;
}