P2866 [USACO06NOV]Bad Hair Day S

一、理解和感悟

1、号码和值都是非常重要的，不要只记录值，用一个数组 $a[i]$ 来保存值，号码就是 $i$ 。
2、每头牛都是在找到了右侧比自己高的牛时，记录右侧第一个比自己高的牛。
3、最终统计时有两种情况，一种是找到了右侧比自己高的牛，另一种是没有找到右侧比自己高的牛。
4、找到了右侧比自己高的牛，那么 $ans+=j-i-1$ ,简单植树问题两棵树之间夹着几棵树。
5、没有找到右侧比自己高的牛，那么就是它向右可以一直看到头，即 $n-i$ 个。

二、完整代码

#include <bits/stdc++.h>

using namespace std;
typedef long long LL;
const int N = 80010;
stack<int> stk;
LL a[N], res[N];
int n;
LL ans;

/**
输入样例
6
10 3 7 4 12 2

找出右侧第一个比自己大的数字
原值：10 3 7 4 12 2
原号：1  2 3 4 5  6

对值：12 7 12 12 -1 -1
对号：5  3 5  5  -1 -1
*/
int main() {
    cin >> n;
    for (int i = 1; i <= n; i++)cin >> a[i];
    for (int i = 1; i <= n; i++) {
        while (!stk.empty() && a[stk.top()] <= a[i]) {
            res[stk.top()] = i;
            stk.pop();
        }
        stk.push(i);
    }
    for (int i = 1; i <= n; i++)
        if (res[i]) ans += res[i] - i - 1;//两个号之间夹的，需要j-i-1,比如4和2之间，其实就是一个3，就是4-2-1=1
        else ans += n - i; //如果右侧没有比自己大的，那么就是一直到最右端，它都能看到。
    //输出结果
    cout << ans << endl;
    return 0;
}