MT3039 山脉

简介: MT3039 山脉

6f2cd0b0d225444ea5de1ac3e27c4d4b.jpg

e1d03777e74848bca043a34db7884f67.jpg

思路:


往右看能看到山顶,可以看成找第一个比当前元素>=的元素,即构造单调递减栈。


例子:


7 5 3 4

1. 7入栈: 7

2. 5入栈: 7 5  ans=ans+1(1是指有1个元素(7)可以看到5)

3. 3入栈: 7 5 3 ans=ans+2(2是指有2个元素(7,5)可以看到3)

4. 3出栈4入栈: 7 5 4 ans=ans+2(2是指有2个元素(7,5)可以看到4)

所以最终ans=1+2+2=5


易错点:


开long long。

判断元素是否进栈需要判断 s.top() <= num,不要忘记等号。

使用 ios::sync_with_stdio(false);   cin.tie(0);  cout.tie(0);在C++中关闭输入输出流的同步,以提高程序的执行效率。

或者使用scanf和printf,避免超时


代码:


#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N = 5e6 + 10;
ll n;
stack<ll> s; // 单调栈
int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);
    cin >> n;
    ll ans = 0;
    ll num;
    for (ll i = 1; i <= n; i++)
    {
        cin >> num;
        while (!s.empty() && s.top() <= num) // 栈顶<=要进来的元素:栈顶元素出栈
        {
            s.pop();
        }
        s.push(num);
        ans += s.size() - 1;
    }
    cout << ans;
    return 0;
}


或者使用scanf和printf

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N = 5e6 + 10;
ll n;
stack<ll> s; // 单调栈
int main()
{
    scanf("%d", &n);
    ll ans = 0;
    ll num;
    for (ll i = 1; i <= n; i++)
    {
        scanf("%d", &num);
        while (!s.empty() && s.top() <= num) // 栈顶<=要进来的元素:栈顶元素出栈
        {
            s.pop();
        }
        s.push(num);
        ans += s.size() - 1;
    }
    printf("%d", ans);
    return 0;
}


相关文章
|
6月前
|
程序员
老程序员分享:MT【202】内准圆
老程序员分享:MT【202】内准圆
25 0
老程序员分享:MT【202】内准圆
|
7月前
|
人工智能 BI
MT3028 正反卡牌
MT3028 正反卡牌
|
7月前
|
人工智能
MT2049 运动会进行中
MT2049 运动会进行中
|
7月前
|
存储
MT3037 新月轩就餐
MT3037 新月轩就餐