Bad Hair Day 单调栈 POJ 3250

原题 :

           Bad Hair Day

题目大意:

   一排牛,从左向右,高的可以看到低的,问总共能看到多少只牛。

   每只奶牛都有一个身高hi(1 ≤ hi ≤ 1,000,000,000),现在在这里有一排全部面向右方的奶牛,一共有N只(1 ≤ N ≤ 80,000)。对于奶牛i来说,如果奶牛i+1,i+2,……,N这些奶牛的身高严格小于奶牛i,则奶牛i可以看到它们凌乱的发型。

输入

第一行:奶牛数量N
第二到N+1行:第i+1行输入奶牛i的身高

10  3  7  4  12  2

输出

一个整数,即他们的和。

5  (3+0+1+1)

解题思路:

   从左向右数值依次递减,栈内数字个数之和即为所求。

#include<cstdio>
#include<algorithm>
#include<stack>
using namespace std;
stack<long long> sta; 
int main(){
	int n;
	scanf("%d",&n);
	long long x,answer;
	scanf("%lld",&x);
	sta.push(x);
	answer = 0;
	for(int i = 1;i < n;i++){
		scanf("%lld",&x);
		while(!sta.empty() && sta.top() <= x){
			sta.pop();  //当栈顶元素小于下一个元素时,将其推出
		}
		answer += sta.size();  //逐个累计
		sta.push(x);   //将下一个元素入栈
	}
	printf("%lld\n",answer);
	return 0;
}

猜你喜欢

转载自blog.csdn.net/error311/article/details/81291515
今日推荐