欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

LeetCode刷题笔记——股票价格跨度

程序员文章站 2022-03-14 23:18:10
...

LeetCode刷题笔记——股票价格跨度
分析题意:当每次调用next函数时,需要从最后一个元素起遍历一遍,直到遇到比当日股票值大为止。
那么我们就能很快地写出第一个版本:

class StockSpanner {
public:
    int next(int price) {
    	int cnt=1;	//因为要算上本身,所以cnt起始值为1而不是0
    	for(size_t i=v.size();i>=0;--i){
    		if(v[i]>price) break;
    		++cnt;
    	}
    	v.push_back(price);
    	return cnt;
    }
private:
    vector<int> v;
};

但是由于时间限制比较严,最后以超时失败告终。
版本一存在很多重复遍历的问题:如果v.back()小于price的话,那么之前数过的cnt个数也都小于price。
为了解决这个问题,我们把比price小的数给折叠起来,相当于建了一个文件夹,文件夹里是小于等于price的数。
这样一来,我们可以从尾往头数,如果尾巴这个数小于等于price,则cnt直接加上尾巴这个文件夹里面数字的个数(相当于合并这个文件夹),再删除位于尾巴这个文件夹,以此类推,直到遇到比price大的数。
由于操作都是从尾巴开始,因此我们可以使用栈来解决,代码如下:

class StockSpanner {
public:
    int next(int price) {
        int cnt=1;
        while(!st1.empty() && st1.top()<=price){
            st1.pop();
            cnt+=st2.top();	//加上st1.top()这个文件夹里的文件数
            st2.pop();
        }
        st1.push(price);
        st2.push(cnt);	//相当于一个大文件夹,里面有多个子文件夹,存储文件夹里共有多少文件
        return cnt;
    }
private:
    stack<int> st1,st2; //st1用来存放文件夹,st2为记录每个文件夹里数字的个数
};
相关标签: LeecCode刷题笔记