今日主要总结一下,LeetCode901. 股票价格跨度(单调栈)
题目:LeetCode901. 股票价格跨度(单调栈)
题目描述:
编写一个 StockSpanner 类,它收集某些股票的每日报价,并返回该股票当日价格的跨度。
今天股票价格的跨度被定义为股票价格小于或等于今天价格的最大连续日数(从今天开始往回数,包括今天)。
例如,如果未来7天股票的价格是 [100, 80, 60, 70, 60, 75, 85],那么股票跨度将是 [1, 1, 1, 2, 1, 4, 6]。
示例:
输入:[“StockSpanner”,“next”,“next”,“next”,“next”,“next”,“next”,“next”],
[[],[100],[80],[60],[70],[60],[75],[85]]
输出:[null,1,1,1,2,1,4,6]
解释:
首先,初始化 S = StockSpanner(),然后:
S.next(100) 被调用并返回 1,
S.next(80) 被调用并返回 1,
S.next(60) 被调用并返回 1,
S.next(70) 被调用并返回 2,
S.next(60) 被调用并返回 1,
S.next(75) 被调用并返回 4,
S.next(85) 被调用并返回 6。
注意 (例如) S.next(75) 返回 4,因为截至今天的最后 4 个价格
(包括今天的价格 75) 小于或等于今天的价格。
一、题目思路分析
题目中的数组[100, 80, 60, 70, 60, 75, 85],我们拿这个分析下:
第一个元素100,它左侧连续小于等于100的只有它自身,所以返回1;
第二个元素80,它左侧连续小于等于80的只有它自身,所以返回1;
第三个元素60,它左侧连续小于等于60的只有它自身,所以返回1;
第四个元素70,它左侧连续小于等于70的只有60和它自身,所以返回2;
第五个元素60,它左侧连续小于等于60的只有它自身,所以返回1;
第六个元素75,它左侧连续小于等于75的有60、70、60和它自身,所以返回4;
第七个元素85,它左侧连续小于等于85的有80、60、70、60、75和它自身,所以返回6;
为了更清楚地分析整个过程,通过画图的方式来说明:
使用文字来描述整个过程:
第一步:100入栈,此时100的跨度仅仅包含自身,跨度为1;
第二步:80入栈,100 > 80,80的跨度仅仅包含自身,跨度为1;
第三步:60入栈,80 > 60,60的跨度仅仅包含自身,跨度为1;
第四步:70入栈,60 < 70,80 > 70,70的跨度包含60和自身,跨度为2(将60的跨度累加起来),此时60应该出栈,因为70后面的元素,如果大于等于70,那么肯定大于60,这个时候70的跨度已经把60包含进去了,所以后面大于等于70的元素,小于80的元素,肯定把60包含进去了,所以60已经没有意义了;如果后面的元素小于70,那么60也起不到作用,因为被70隔开了;
第五步:60入栈,此时60的跨度仅仅包含自身,跨度为1;
第六步:75入栈,此时60和70都应该出栈,道理和第四步一样,75的跨度为4,分别累加了70的跨度2和60的跨度1,再加上自身1,所以跨度是4;
第七步:85入栈,此时75和80都应该出栈,道理和第四步一样,75的跨度为4,80的跨度为1,再加上自身,所以跨度是6。
本题重难点
单调栈要同时存储price和当前price的跨度要想到使用pair类型数据结构
二、C++实现代码
class StockSpanner {
public:
stack<pair<int, int>> st;
StockSpanner() {
}
int next(int price) {
pair<int, int> cur(price, 1);
while(!st.empty() && cur.first >= st.top().first){
cur.second += st.top().second;
st.pop();
}
st.push(cur);
return cur.second;
}
};
/**
* Your StockSpanner object will be instantiated and called as such:
* StockSpanner* obj = new StockSpanner();
* int param_1 = obj->next(price);
*/