您的位置:首页 > 大数据 > 人工智能

POJ 3250 Bad Hair Day(单调栈)

2015-07-22 12:09 309 查看
用一个栈来维护当前的递减序列,从左到右遍历,对于每个数cur,从栈顶开始把所有小于等于cur的数字都出栈,然后剩下的就都是可以看到cur的牛,并且保证了栈中的单调性。每次遍历对弹栈后的长度求和即可。

代码:

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cstring>
using namespace std;
#define LL long long
int N;
int q[80005];
int top;
int main(){
    scanf("%d",&N);
    top=0;
    int cur;
    LL res=0;
    for(int i=0;i<N;i++){
        scanf("%d",&cur);
        while(top&&q[top]<=cur){
            top--;
        }
        res+=top;
        q[++top]=cur;
    }
    printf("%I64d\n",res);
    return 0;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: