leetcode - 每日一题 - 901. 股票价格跨度

100 阅读1分钟

持续创作,加速成长!这是我参与「掘金日新计划 · 10 月更文挑战」的第6天,点击查看活动详情


901. 股票价格跨度

题目:

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

提示:

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

个人思路

1.简单暴力解(超出时间限制。。。)

定义两个栈,栈A和栈B

栈A负责存入每日报价,执行查询操作时,将栈A的数据弹出,存入栈B,并计数。 当发现数据比查询数据大时,结束。 再将栈B的数据存回栈A。

class StockSpanner {
        std::stack<int> a,b;

public:
    StockSpanner() {
    }
    
    int next(int price) {
        a.push(price);
        int spanner = 0;
        while(!a.empty()&&a.top() <= price){
            b.push(a.top());
            a.pop();
            spanner++;
        }

        while(!b.empty()){
            a.push(b.top());
            b.pop();
        }

        return spanner;
    }
};

/**
 * Your StockSpanner object will be instantiated and called as such:
 * StockSpanner* obj = new StockSpanner();
 * int param_1 = obj->next(price);
 */

结果。。。太暴力,超时了。。。

image.png 测试用例通过86/99

2、暴力解(过了)

改用数组vector 向数组中插入元素,从最后一个元素向前遍历,判断数值是否小于等于当前传入值,累计计数。

class StockSpanner {
        std::vector<int> stack;

public:
    StockSpanner() {
    }
    
    int next(int price) {
        int spanner = 0;

        stack.push_back(price);
        for(int i = stack.size();i>0;i--){
            if(stack[i-1]<=price){
                spanner++;
            }else{
                return spanner;
            }
        }

        return spanner;
    }
};

/**
 * Your StockSpanner object will be instantiated and called as such:
 * StockSpanner* obj = new StockSpanner();
 * int param_1 = obj->next(price);
 */

image.png