1. 程式人生 > >[BZOJ1595] [Usaco2008 Jan]人工湖(單調棧)

[BZOJ1595] [Usaco2008 Jan]人工湖(單調棧)

入棧 tchar fin clas geo getc main pri include

傳送門

好難的題。。至少對我來說。

這題就是模擬從最低的平臺註水,然後將最低的填滿以後從最低的平臺向兩邊擴展,每次找最近的最低的平臺h,然後將水填到h高度。 棧裏存的是向外擴展的時候,有時會遇到高度遞減的情況,這時並不能填水,但要把這些高度都遞減(即擴展時的順序)記錄進棧,然後遇到一個比比水面高的平臺h時,模擬倒水,水會挨個淹沒最低的平臺,即需要從棧頂一個一個出戰計算淹沒時間,直至棧頂平臺高度>h,此時h入棧。重復執行就可算出答案。

#include <cstdio>
#include <iostream>
#define N 100011
#define INF 1000011
#define LL long long
#define min(x, y) ((x) < (y) ? (x) : (y))

int s[N];
int n, p = 1, top;
LL t, sum[N], h[N], w[N], add[N], ans[N];

inline LL read()
{
	LL x = 0, f = 1;
	char ch = getchar();
	for(; !isdigit(ch); ch = getchar()) if(ch == ‘-‘) f = -1;
	for(; isdigit(ch); ch = getchar()) x = (x << 1) + (x << 3) + ch - ‘0‘;
	return x * f;
}

int main()
{
	int i, l, r;
	n = read();
	h[0] = h[n + 1] = INF;
	for(i = 1; i <= n; i++)
	{
		sum[i] = sum[i - 1] + read();
		h[i] = read();
		if(h[p] > h[i]) p = i;
	}
	for(i = 1; i <= n + 1; i++)
	{
		while(top && h[s[top]] < h[i])
		{
			w[s[top]] = sum[i - 1] - sum[s[top - 1]];
			add[s[top]] = w[s[top]] * (min(h[s[top - 1]], h[i]) - h[s[top]]);
			top--;
		}
		s[++top] = i;
	}
	top = 0;
	l = r = s[++top] = p;
	for(i = 1; i <= n; i++)
	{
		if(h[l - 1] < h[r + 1]) p = --l;
		else p = ++r;
		while(top && h[s[top]] < h[p])
		{
			ans[s[top]] = t + w[s[top]];
			t += add[s[top]];
			top--;
		}
		s[++top] = p;
	}
	for(i = 1; i <= n; i++) printf("%lld\n", ans[i]);
	return 0;
}

  

[BZOJ1595] [Usaco2008 Jan]人工湖(單調棧)