LeetCode901 - 股票价格的跨度

LeetCode901 - 股票价格的跨度

链接：https://leetcode-cn.com/problems/online-stock-span
编写一个 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) 小于或等于今天的价格。

解法一：时间复杂度O(n^2)，空间复杂度O(n)，n是调用next次数

class StockSpanner {

    List<Integer> list;

    public StockSpanner() {
        list = new ArrayList<Integer>();
    }
    
    public int next(int price) {
        list.add(price);
        int ret = 1;
        if(list.size() == 1) {
            return ret;
        }
        for(int i=list.size()-2; i>=0; i--) {
            if(list.get(i) > list.get(list.size()-1)) {
                break;
            }
            ret++;
        }
        return ret;
    }
}

解法二、单调栈，维护两个栈，一个负责实时添加price，一个用于记录在前一个数之前比前一个小的数个数，核心在于实时添加有返回值

时间复杂度O(n)，空间复杂度O(n)，n为调next函数的次数

class StockSpanner {

    Stack<Integer> prices, cnts;

    public StockSpanner() {
        prices = new Stack<Integer>();
        cnts = new Stack<Integer>();
    }
    
    public int next(int price) {
        int cnt = 1;
        while(!prices.isEmpty() && prices.peek() <= price) {
            prices.pop();
            cnt += cnts.pop();
        }
        prices.push(price);
        cnts.push(cnt);
        return cnt;
    }
}