leetcode 901:力扣
今天做这道题时,第一次了解到单调栈这种方法,就是用栈维护一个单调递增或单调递减的序列
本题,需要找到的是一个股票价格的跨度,常规想法就是从后往前查找,对小于等于当前股票价格的连续天数计数,但这个时间复杂度会比较高,应该是O(n^2)
考虑,我们从当前股票价格A[i]往前查找的过程中,每次都是遇到第一个A[j]比当前股票价格大的就停止了,中间这些j<k<=i都是小于等于当前股票价格的,也就没有必要记录,只记录个数count即可;如果下一个股票价格A[i+1]大于当前这个股票价格A[i],加上当前计数count即可;如果A[i+1]<A[i],因为要求连续,前面小于的值也与A[i+1]无关
所以,我们用栈维护一个股票价格单调递减的序列,同时每个股票价格有个count值,是当前这个股票价格最大的跨度。每次遍历到一个A[i]时,从栈顶开始比较,如果小于等于当前A[i]就弹出,并加上它的count值,如果大于当前A[i]就退出循环,把A[i]的count值加1压入栈顶。
int next(int price) {
int count = 1;
while (!stock.empty()) {
pair<int, int> temp = stock.top();
if (stock.top().first <= price) {
count += stock.top().second;
stock.pop();
} else {
break;
}
}
stock.push(make_pair(price, count));
return count;
}