每日题解:LeetCode 901. 股票价格跨度

Author Avatar
清水雅然君 08月 07,2020

题目地址

题目描述

编写一个 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) 小于或等于今天的价格。

提示:

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

解法

class StockSpanner {

       Stack<Integer> prices, weights;
    public StockSpanner() {
        prices = new Stack();
        weights = new Stack();
    }

    public int next(int price) {
        int w=1;
        while (!prices.isEmpty() && prices.peek()<=price){
            w+=weights.pop();
            prices.pop();
        }
        weights.push(w);
        prices.push(price);
        return w;
    }
}

java

解题思路

单调栈

分析题目要求,假设输入的price,存入一个数组 [100, 80, 60, 70, 60, 75, 85],比如元素,之前比它小的元素是60,则70的值应该是2(包括今天),依次类推,75之前比它的是3个,值就是4
最终答案输出的值为[1,1,1,2,1,4,6],维护一个单调递减的栈
1、第一个元素100,此时压入栈,同时记录下之前比它小的元素次数为1;这里维护两个栈
2、第一个元素80,第三个元素60,之前都没比他们下的元素;prices栈内 100,80,60,weights内1,1,1
3、遍历70,由于prices栈顶元素比70小,弹出栈顶元素,同时记录70的次数为2
4、60,栈顶元素为75,记录 prices栈内 100,80,70,60,weights内1,1,2,1
5、75,由于75比栈顶元素大,依次弹出直到栈顶元素比75大,此时 prices栈内 100,80,75,weights内1,1,4
6、同理,遍历80元素时,维护栈的元素为prices栈内 100,85,weights内1,6
所以这题的解法为

 public int next(int price) {
        int w=1;
        while (!prices.isEmpty() && prices.peek()<=price){
        //遍历当前的元素小的次数
            w+=weights.pop();
            //弹出元素
            prices.pop();
        }
        weights.push(w);
        prices.push(price);
        return w;
    }