POJ3250 牛的视野(单调栈)

一群高度不完全相同的牛从左到右站成一排,每头牛只能看见它右边的比它矮的牛的发型,若遇到一头高度大于或等于它的牛,则无法继续看到这头牛后面的其他牛。

给出这些牛的高度,要求每头牛可以看到的牛的数量的和。

把要求作一下转换,其实就是要求每头牛被看到的次数之和。这个可以使用单调栈来解决。

从左到右依次读取当前牛的高度,从栈顶开始把高度小于或等于当前牛的高度的那些元素删除,此时栈中剩下的元素的数量就是可以看见当前牛的其他牛的数量。把这个数量加在一起,就可以得到最后的答案了。

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

const int N = 80005;
int stack[N];
int top, n;

int main()
{
	int num;
	long long ans = 0;
	scanf("%d", &n);
	for (int i = 0; i < n; ++i)
	{
		scanf("%d", &num);
		while (top > 0 && stack[top - 1] <= num) --top;
		ans += top;
		stack[top++] = num;
	}
	printf("%lld\n", ans);
	return 0;
}


你可能感兴趣的:(POJ3250 牛的视野(单调栈))