思路:
往右看能看到山顶,可以看成找第一个比当前元素>=的元素,即构造单调递减栈。
例子:
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; }