[lnsyoj102/luoguP2866]Bad Hair Day
题意
给定序列 ,记 为 右侧第一个 的元素与 间的元素个数,求
sol
单调栈可以在 的时间复杂度内解决求某个元素左(右)第一个大于(小于)的元素。
以本题为例,由于本题需要求右侧第一个 的元素,因此需要维护一个严格单调递减栈。从 到 枚举,当枚举到 时,将栈顶的所有 的元素都弹出栈,这样就可以保证栈内的元素永远严格单调递减。同时,因为所有 的元素都已出栈,因此当前的栈顶元素就是满足上述题意的元素
注意:由于需要利用下标求 ,因此栈内应存元素的下标而非元素的值
代码
#include <iostream>
#include <algorithm>
#include <cstring>
using namespace std;
const int N = 80005;
int a[N];
int n;
int stk[N], top;
int main(){
scanf("%d", &n);
for (int i = 1; i <= n; i ++ ) scanf("%d", &a[i]);
long long res = 0;
stk[0] = n + 1;
for (int i = n; i >= 1; i -- ){
while (top && a[stk[top]] < a[i]) top -- ;
res += stk[top] - i - 1;
stk[ ++ top] = i;
}
printf("%lld\n", res);
return 0;
}
分类:
题解 / 2024训练
标签:
数据结构
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· 分享一个免费、快速、无限量使用的满血 DeepSeek R1 模型,支持深度思考和联网搜索!
· 使用C#创建一个MCP客户端
· 基于 Docker 搭建 FRP 内网穿透开源项目(很简单哒)
· ollama系列1:轻松3步本地部署deepseek,普通电脑可用
· 按钮权限的设计及实现