设计一个算法收集某些股票的每日报价,并返回该股票当日价格的 跨度 。
当日股票价格的 跨度 被定义为股票价格小于或等于今天价格的最大连续日数(从今天开始往回数,包括今天)。
例如,如果未来 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 <= 10^5
最多调用 next 方法 10^4 次
思路:
单调栈思路。
定义单调栈 price_stack, 以价格生成非递减栈,栈中记录天数 index 与 price 价格
1. 若是栈为空,那么说明这是第一天的价格,返回 1 ,并将 index = 0 与 price 压入栈中
2. 若是当前 price 大于栈顶元素的价格,那么将栈顶元素弹出栈,直到栈为空或者栈顶元素的价格大于当前 price 位置 为止
2.1 如果 最后栈为空,说明以前的每一天的价格都小于当期 price,那么 返回 index + 1 即可,同时将当前的 index 与price压入栈中
2.2 如果栈不为空,说明当前栈顶元素的价格大于 当前价格 price,那么题目中所求的最大连续天数为 index - stack.top[ 0 ]
c++:
class StockSpanner {
public:
stack<vector<int>> price_stack;
int index = 0; // 第 index 天获取价格
StockSpanner() {
}
int next(int price) {
if(price_stack.empty()) {
vector<int> vec; // vec[0] 为第几天获取价格,从第 0 天开始
vec.push_back(index++);
vec.push_back(price);
price_stack.push(vec);
return 1;
} else {
// 找到上一个大于 price 的那天,若是栈中的价格都小于 price 的话,那么清空栈中元素
while(!price_stack.empty() && price_stack.top()[1] <= price) {
price_stack.pop();
}
int result = 1;
// 若是栈中位空,说明 栈中的价格都小于 price
if(price_stack.empty()) {
result = index + 1;
} else {
// 当前价格的天数减去栈顶元素的天数
result = index - price_stack.top()[0];
}
vector<int> vec; // vec[0] 为第几天获取价格,从第 0 天开始
vec.push_back(index++);
vec.push_back(price);
price_stack.push(vec);
return result;
}
}
};
/**
* Your StockSpanner object will be instantiated and called as such:
* StockSpanner* obj = new StockSpanner();
* int param_1 = obj->next(price);
*/