[leetcode/lintcode 题解] 算法面试高频题详解:股票价格跨度

描述
编写一个 StockSpanner 类,它收集某些股票的每日报价,并返回该股票当日价格的跨度。
今天股票价格的跨度被定义为股票价格小于或等于今天价格的最大连续日数(从今天开始往回数,包括今天)。
例如,如果未来7天股票的价格是 [100, 80, 60, 70, 60, 75, 85],那么股票跨度将是 [1, 1, 1, 2, 1, 4, 6]。

  • 调用 StockSpanner.next(int price) 时,将有 1 <= price <= 10^5。
  • 每个测试用例最多可以调用 10000 次 StockSpanner.next。
  • 在所有测试用例中,最多调用 150000 次 StockSpanner.next。
  • 此问题的总时间限制减少了 50%。

在线评测地址:领扣题库官网

样例1
输入:prices = [100,80,60,70,60,75,85]
输出:[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) 小于或等于今天的价格。

样例2
输入:prices = [50,80,80,70,90,75,85]
输出:[1,2,3,1,5,1,2]
解释:
首先,初始化 S = StockSpanner(),然后:
S.next(50) 被调用并返回 1,
S.next(80) 被调用并返回 2,
S.next(80) 被调用并返回 3,
S.next(70) 被调用并返回 1,
S.next(90) 被调用并返回 5,
S.next(75) 被调用并返回 1,
S.next(85) 被调用并返回 2。

解题思路
单调栈问题 题目中提到股票价格小于或等于今天价格的最大连续日数。 由于这是一个在线问题,所以我们必然是要将输入的price给存储起来,而且同时我们也需要保留这是第几次输入的信息。 需要注意的是边界问题,当我们输入第一个price的时候,此时stack空。 所以这里拿出来判断特殊处理一下
源代码

public class StockSpanner {
    public StockSpanner() {
        
    }
    
    /**
     * @param price: 
     * @return: int
     */
     
     Stack<int[]> stack = new Stack<>();
    public int next(int price) {
        // Write your code here.
        int res = 1;
        while (!stack.isEmpty() && stack.peek()[0] <= price)
            res += stack.pop()[1];
        stack.push(new int[]{price, res});
        return res;
    }
}

更多题解参考:九章官网solution

上一篇:[leetcode/lintcode 题解] 阿里巴巴面试题: 路径和 IV


下一篇:IoT离线设备云端控制指令消息触达方案