901 股票价格跨度——Leetcode天天刷(20022.10.21)【单调栈】
前言
刚写了今天的每日一题,有点难,一开始没知道应该用单调栈,但是不清楚怎么用,只能说还是学艺不精吧,找时间需要练练算法了。
看了题解才比较理解。
题目简述
就是一个StockSpanner类,除了构造函数外,还有next函数,是求当前价格往前不大于当前价格的最大连续天数。
题目链接:传送门
吐槽
看这个题目描述,我觉得我想不出怎么用单调栈有理由了,它这个最大连续天数其实根本不准确,应该说是往前数第一个比当前价格大的日子的中间天数,如果是最大连续天数,其实不一定是从当天开始的连续天的价格序列,所以,出题人的问题,黄同学不背![傲娇脸]
本地调试
- 环境:
devc++
,mingw
环境即可。 - 输入,多组输入,每组输入一个整数n,表示要调用多少次
next
函数,然后输入n个整数。 - 输出,对应输入,每调用一次
next
,输出一个整数。
解法 单调栈C++
真的是越来越笨了,单调栈都不会写!
看到我的吐槽,是不是清楚了这道题目的描述了,我们只要逆着找到第一个比当前价格高的日子,然后中间的天数就是答案。
相信聪明如你们,肯定想到了,最直接的方法,那不就二重循环前向遍历嘛,这有什么难的?
但是吧,这种暴力的思路确实很容易写,但是时间复杂度就达到了恐怖的 O ( n 2 ) O(n^2) O(n2),所以,在这里,黄同学想介绍一种很常见却很好用的解法——单调栈。
我相信肯定知道啥是栈,那么单调栈就是里边的元素是单调的,对,就是我们在数学里边学的函数或者数列单调递增/递减 的单调。
我们可以用一个单调栈 stack<pair<int, int>>
里边的元素其实就是 {price, days}
,表示的就是价格还有距离上一个比当前大的中间天数,根据题目最小是1。
这个单调栈是一个递减的,如果栈顶价格低于当前价格,那么出栈,并且中间天数对应加上;否则就入栈。
复杂度分析与执行效率
时间复杂度: O ( n ) O(n) O(n),其实最多也就相当于遍历两遍序列,所以线性级的复杂度。
空间复杂度: O ( n ) 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;
}
后话
摆太久了,立个flag,一周2+篇blog,且至少1篇不是刷题这种的。