[lnsyoj102/luoguP2866]Bad Hair Day

题意

给定序列 a,记 Ciai 右侧第一个 ai 的元素与 ai 间的元素个数,求i=1nCi

sol

单调栈可以在 O(n) 的时间复杂度内解决求某个元素左(右)第一个大于(小于)的元素。
以本题为例,由于本题需要求侧第一个 ai 的元素,因此需要维护一个严格单调递减栈。从 n1 枚举,当枚举到 ai 时,将栈顶的所有 <ai 的元素都弹出栈,这样就可以保证栈内的元素永远严格单调递减。同时,因为所有 <ai 的元素都已出栈,因此当前的栈顶元素就是满足上述题意的元素
注意:由于需要利用下标求 Ci,因此栈内应存元素的下标而非元素的值

代码

#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;
}
posted @   是一只小蒟蒻呀  阅读(11)  评论(0编辑  收藏  举报
相关博文:
阅读排行:
· 分享一个免费、快速、无限量使用的满血 DeepSeek R1 模型,支持深度思考和联网搜索!
· 使用C#创建一个MCP客户端
· 基于 Docker 搭建 FRP 内网穿透开源项目(很简单哒)
· ollama系列1:轻松3步本地部署deepseek,普通电脑可用
· 按钮权限的设计及实现
点击右上角即可分享
微信分享提示