POJ 3250 Bad Hair Day(单调栈)

最新推荐文章于 2020-05-06 21:34:07 发布

小爷永远不死

最新推荐文章于 2020-05-06 21:34:07 发布

阅读量604

点赞数

分类专栏：数据结构与算法

本文链接：https://blog.csdn.net/fobdddf/article/details/26986003

版权

数据结构与算法专栏收录该内容

338 篇文章 0 订阅

订阅专栏

题目链接：POJ 3250 Bad Hair Day

只需要向前统计，不需要向后统计，比前两道题还简单些。

#include <iostream>
#include <cstdio>
#include <stack>

using namespace std;

struct Node
{
    int l;
    long long h;
};

int main()
{
    int n;
    scanf("%d", &n);
    long long sum = 0;
    stack <Node> S;
    Node temp, tmp;
    for(int i = 0; i < n; i++)
    {
        scanf("%lld", &temp.h);
        temp.l = i;
        while(!S.empty() && S.top().h <= temp.h)
        {
            tmp = S.top();
            S.pop();
            sum += i - tmp.l - 1;
            //temp.l = tmp.l;
        }
        S.push(temp);
    }
    while(!S.empty())
    {
        tmp = S.top();
        S.pop();
        sum += n - tmp.l - 1;
    }
    printf("%lld\n", sum);
    return 0;
}