本文涉及的基础知识点
C++单调栈
LeetCode901. 股票价格跨度
设计一个算法收集某些股票的每日报价,并返回该股票当日价格的 跨度 。
当日股票价格的 跨度 被定义为股票价格小于或等于今天价格的最大连续日数(从今天开始往回数,包括今天)。
例如,如果未来 7 天股票的价格是 [100,80,60,70,60,75,85],那么股票跨度将是 [1,1,1,2,1,4,6] 。
实现 StockSpanner 类:
StockSpanner() 初始化类对象。
int next(int price) 给出今天的股价 price ,返回该股票当日价格的 跨度 。
示例:
输入:
[“StockSpanner”, “next”, “next”, “next”, “next”, “next”, “next”, “next”]
[[], [100], [80], [60], [70], [60], [75], [85]]
输出:
[null, 1, 1, 1, 2, 1, 4, 6]
解释:
StockSpanner stockSpanner = new StockSpanner();
stockSpanner.next(100); // 返回 1
stockSpanner.next(80); // 返回 1
stockSpanner.next(60); // 返回 1
stockSpanner.next(70); // 返回 2
stockSpanner.next(60); // 返回 1
stockSpanner.next(75); // 返回 4 ,因为截至今天的最后 4 个股价 (包括今天的股价 75) 都小于或等于今天的股价。
stockSpanner.next(85); // 返回 6
提示:
1 <= price <= 105
最多调用 next 方法 104 次
单调栈
从小到大处理第i天,从i-1到0,寻找第一天股价大于今天,令其为j,则第i天最长连续日为i-j。
处理price[i]时:
i1 < i2 < i,如果price[i1] < price[i2] 则i1被淘汰i2淘汰。淘汰后,价格严格递减。
淘汰后,栈顶元素就是j,如果栈顶为空,j就是-1。
代码
核心代码
class StockSpanner {
public:
StockSpanner() {
}
int next(int price) {
while (m_sta.size() && (price >= m_nums[m_sta.top()])) {
m_sta.pop();
}
const int left = m_sta.empty() ? -1 : m_sta.top();
m_sta.emplace(m_nums.size());
m_nums.emplace_back(price);
return m_nums.size() - 1 - left;
}
vector<int> m_nums;
stack<int> m_sta;
};
单元测试
vector<int> nums;
TEST_METHOD(TestMethod11)
{
nums = {100,80,60,70,60,75,85};
StockSpanner stockSpanner;
vector<int> res;
for (const auto& n : nums) {
res.emplace_back(stockSpanner.next(n));
}
AssertEx(vector<int>{1, 1, 1, 2, 1, 4, 6}, res);
}