编程语言
首页 > 编程语言> > 【算法】POJ 3250 Bad Hair Day

【算法】POJ 3250 Bad Hair Day

作者:互联网

解题思路

将每头牛加入单调栈,记录每次pop操作后栈的长度,也就是每头牛能被多少牛看到,累加之后与每头牛能看到的牛的数量之和一致。
题目里\(1<=n<=80,000\),如果牛的身高序列单调递减,结果最大为\(79999+79998+...+1=79999*80000/2\approx 3.2*10^9\),超出了int的范围,需要用long long

AC代码

#include <iostream>
#include <stack>

int main() {
    int n;
    std::cin >> n;
    std::stack<int> stack;
    int cow;
    std::cin >> cow;
    stack.push(cow);
    long long res = 0;
    for (int i = 1; i < n; i++) {
        std::cin >> cow;
        while (!stack.empty() && stack.top() <= cow) {
            stack.pop();
        }
        res += stack.size();
        stack.push(cow);
    }
    std::cout << res << std::endl;
    return 0;
}

标签:std,cow,3250,cin,long,Hair,int,Bad,stack
来源: https://www.cnblogs.com/huzheyu/p/algorithm-poj-3250.html