题目地址:
https://www.lintcode.com/problem/online-stock-span/description
要求设计一个类,可以实现一个操作,添加一个数,并返回它之前添加的小于等于该数的连续个数。
思路是单调栈。开一个单调上升栈,如果栈空或者栈顶的数字小于等于新来的数字,则将新数字连同其下标入栈(这里由于是数据流,并没有下标,但可以开个变量记录下标);否则pop栈顶,直到栈空或者栈顶大于新来的数;满足条件的数的个数就是新来的数的下标减去栈顶的下标(栈空则视栈顶的下标是 − 1 -1 −1)。代码如下:
import java.util.ArrayDeque;
import java.util.Deque;
public class StockSpanner {
private Deque<int[]> stack;
private int idx;
public StockSpanner() {
stack = new ArrayDeque<>();
}
/**
* @param price:
* @return: int
*/
public int next(int price) {
// Write your code here.
while (!stack.isEmpty() && stack.peek()[1] <= price) {
stack.pop();
}
int prev = stack.isEmpty() ? -1 : stack.peek()[0];
stack.push(new int[]{idx++, price});
return idx - prev - 1;
}
}
时空复杂度 O ( n ) O(n) O(n)。