901 股票价格跨度——Leetcode天天刷（20022.10.21）【单调栈】

文章目录

901 股票价格跨度——Leetcode天天刷（20022.10.21）【单调栈】

前言

刚写了今天的每日一题，有点难，一开始没知道应该用单调栈，但是不清楚怎么用，只能说还是学艺不精吧，找时间需要练练算法了。

看了题解才比较理解。

题目简述

就是一个StockSpanner类，除了构造函数外，还有next函数，是求当前价格往前不大于当前价格的最大连续天数。

题目链接：传送门

吐槽

看这个题目描述，我觉得我想不出怎么用单调栈有理由了，它这个最大连续天数其实根本不准确，应该说是往前数第一个比当前价格大的日子的中间天数，如果是最大连续天数，其实不一定是从当天开始的连续天的价格序列，所以，出题人的问题，黄同学不背！[傲娇脸]

本地调试

环境：devc++ ，mingw环境即可。
输入，多组输入，每组输入一个整数n，表示要调用多少次 next函数，然后输入n个整数。
输出，对应输入，每调用一次next，输出一个整数。

解法单调栈C++

真的是越来越笨了，单调栈都不会写！

看到我的吐槽，是不是清楚了这道题目的描述了，我们只要逆着找到第一个比当前价格高的日子，然后中间的天数就是答案。

相信聪明如你们，肯定想到了，最直接的方法，那不就二重循环前向遍历嘛，这有什么难的？

但是吧，这种暴力的思路确实很容易写，但是时间复杂度就达到了恐怖的 $O(n^2)$ ，所以，在这里，黄同学想介绍一种很常见却很好用的解法——单调栈。

我相信肯定知道啥是栈，那么单调栈就是里边的元素是单调的，对，就是我们在数学里边学的函数或者数列单调递增/递减 的单调。

我们可以用一个单调栈 stack<pair<int, int>> 里边的元素其实就是 {price, days}，表示的就是价格还有距离上一个比当前大的中间天数，根据题目最小是1。

这个单调栈是一个递减的，如果栈顶价格低于当前价格，那么出栈，并且中间天数对应加上；否则就入栈。

复杂度分析与执行效率

时间复杂度： $O (n)$ ，其实最多也就相当于遍历两遍序列，所以线性级的复杂度。

空间复杂度： $O (n)$ ，最多一直存，存完整个序列。

在这里插入图片描述

Code（C++）

#include<bits/stdc++.h>

using namespace std;

class StockSpanner 
{
private:
	stack<pair<int, int>> stk;		// 单调栈，每个元素前者是价格，后者是距离第一个比当前价格高的日子的中间天数
	
public:
    StockSpanner() 
	{
    }
    // 单调栈
    int next(int price) 
	{
		// 如果栈空
		if(stk.empty())
		{
			stk.push({price, 1});		// 直接压入价格和天数为1
			return 1;
		}
		int ans = 1;		// 既是答案也是被压入的中间天数
		// 单调栈，如果栈顶的价格不大于当前价格，则出栈
		while(!stk.empty() && stk.top().first <= price)
		{
			ans += stk.top().second;		// 天数累加
			stk.pop();
		}
		stk.push({price, ans});			// 入栈
		return ans;
    }
};

int main()
{
	int n;
	while(~scanf("%d", &n))
	{
		StockSpanner sto;
		int price;
		while(n--)
		{
			scanf("%d", &price);
			printf("%d\n", sto.next(price));
		}
		printf("\n");
	}
	

	return 0;
}