洛谷题单指南-常见优化技巧-P2866 [USACO06NOV] Bad Hair Day S
原题链接:https://www.luogu.com.cn/problem/P2866
题意解读:每个牛能看到的右边比他矮的牛,直到有比他高的挡住为止,因此只用找每个牛右边第一个比他高的牛的位置即可计算中间比他矮的有多少。
解题思路:
典型的单调栈应用,注意,常规的单调栈可以用来:
1、找每个数左边第一个比他小的数的位置
2、找每个数左边第一个比他大的数的位置
但是此题是要找每个牛右边第一个比他矮的,也就是每个数右边第一个比他大的位置,只需要从后往前处理(i从n->1),即可转换为常规的单调栈模型:
第一步:如果栈不空且当前元素大于栈顶元素,则不断弹栈
第二步:此时如果栈不空,栈顶即第一个比当前元素大的数的位置r=s.top();如果栈空可以设该位置为r=n+1;这样当前数能看到的牛数量为r-i-1
第三步:将当前数的位置i入栈
100分代码:
#include <bits/stdc++.h>
using namespace std;
const int N = 80005;
stack<int> s;
int n;
int a[N];
long long ans;
int main()
{
cin >> n;
for(int i = 1; i <= n; i++) cin >> a[i];
for(int i = n; i >= 1; i--)
{
while(s.size() && a[i] > a[s.top()]) s.pop();
//a[i]右边第一个比a[i]大的数的位置
int r = s.empty()? n + 1 : s.top();
ans += r - i - 1;
s.push(i);
}
cout << ans;
return 0;
}
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】凌霞软件回馈社区,博客园 & 1Panel & Halo 联合会员上线
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】博客园社区专享云产品让利特惠,阿里云新客6.5折上折
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步