单调栈 （PTA）（洛谷P1823 音乐会的等待 ）

单调栈 (pintia.cn)

P1823 [COI2007] Patrik 音乐会的等待 - 洛谷 | 计算机科学教育新生态 (luogu.com.cn)

题目描述

N人们排队等着参加音乐会。人们等得很无聊，于是他们转身去排队寻找熟悉的人。
如果两个人A和B并排站在一起，或者如果他们中间没有人比A或B高，那么他们可以看到对方。

编写一个程序，确定可以看到彼此的成对人数。

输入格式:

第一行输入包含一个整数N（1≤N≤500000），排队的人数。

以下N行中的每一行都包含一个整数，即一个人的身高（以纳米为单位）。

每个人的身高都将小于231纳米。高度是按照人们排队的顺序给出的。

输出格式:

输出可以看到对方的成对人数

输入样例:

7
2
4
1
2
2
5
1

输出样例:

10

 

注意：两个一样高的人是可以相互看见的
1 1 1 1 1
这串数字里面
第一个1是可以和最后一个1相互看见的

思路：

此题复杂度用暴力显然会TLE，因此我们可以维护一个单调栈，

单调栈是一个具有单调性的栈，要一直保持单调，遇到不单调的，要删除或覆盖。

记录我们已经找到过的高的人，已经判断操作过的没用的则可以出栈了，栈顶的人最矮，...

要AC掉这道题我们还要考虑两个人身高相同的情况。我们可以在栈里存pair<身高，人数>并维护它，记录下各身高的人数，最后在计算的时候方便处理

如果栈里面的栈顶元素比手上这个元素要小
那栈顶元素左边能够看到的最后一个元素就是手上这个了
再也看不到别的了
这就是1对可以互相看见
弹出就好了
然后手上这个继续和栈顶去比较

题解：

#include <iostream>
#include <stack>
using namespace std;
int main()
{
    long long res = 0;    //答案开大点
    stack<pair<int,int>> st;
    int n,h;    //人数和身高
    cin >> n;
    for(int i = 0; i < n; i++) 
    {
        cin >> h;
        pair<int,int> p; // 将要入栈的元素
        p.first = h;    //存放当前元素身高
        p.second = 1;    //当前元素数量，（为了应对相同身高人的刷新人数操作）

        while(!st.empty() && st.top().first <= h)
        {
            //答案加栈顶元素的数量
            res += st.top().second;
            //进行身高相等的特判
            if(h == st.top().first)
                //如果该元素与A的高度相同,那么与A高度相同的人的个数要加上该元素的个数 
                p.second += st.top().second;
            //出栈
            st.pop();
        }
        如果栈中还有元素，则头元素是第一个大于A的高度的人，也可以和A互相看到 
        if(!st.empty())    res++;
        //将当前元素入栈
        st.push(p);   
    }

    cout << res;

    return 0;
}