POJ 3250 Bad Hair Day(单调栈)

用一个栈来维护当前的递减序列,从左到右遍历,对于每个数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;
}


你可能感兴趣的:(单调栈)