洛谷题单指南-常见优化技巧-P2866 [USACO06NOV] Bad Hair Day S

原题链接：https://www.luogu.com.cn/problem/P2866

题意解读：每个牛能看到的右边比他矮的牛，直到有比他高的挡住为止，因此只用找每个牛右边第一个比他高的牛的位置即可计算中间比他矮的有多少。

解题思路：

典型的单调栈应用，注意，常规的单调栈可以用来：

1、找每个数左边第一个比他小的数的位置

2、找每个数左边第一个比他大的数的位置

但是此题是要找每个牛右边第一个比他矮的，也就是每个数右边第一个比他大的位置，只需要从后往前处理（i从n->1），即可转换为常规的单调栈模型：

第一步：如果栈不空且当前元素大于栈顶元素，则不断弹栈

第二步：此时如果栈不空，栈顶即第一个比当前元素大的数的位置r=s.top()；如果栈空可以设该位置为r=n+1；这样当前数能看到的牛数量为r-i-1

第三步：将当前数的位置i入栈

100分代码：

#include <bits/stdc++.h>
using namespace std;

const int N = 80005;
stack<int> s;
int n;
int a[N];
long long ans;

int main()
{
    cin >> n;
    for(int i = 1; i <= n; i++) cin >> a[i];

    for(int i = n; i >= 1; i--)
    {
        while(s.size() && a[i] > a[s.top()]) s.pop();

        //a[i]右边第一个比a[i]大的数的位置
        int r = s.empty()? n + 1 : s.top();
        ans += r - i - 1;

        s.push(i);
    }

    cout << ans;
    return 0;
}