POJ 3250 Bad Hair Day(单调栈)

原创于 2014-05-25 22:33:31 发布 · 627 阅读

0 ·

CC 4.0 BY-SA版权

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

338 篇文章

订阅专栏

本文提供了一种解决POJ3250 Bad Hair Day问题的有效算法。通过使用栈来跟踪节点，并仅向前统计，实现了高效计算。文章包含了完整的C++代码实现。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目链接：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;
}