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;
}

三、视频讲解

https://www.ixigua.com/6925713196787761677

原文地址:https://www.cnblogs.com/littlehb/p/15262103.html