LeetCode刷题笔记——股票价格跨度
程序员文章站
2022-03-14 23:18:10
...
分析题意:当每次调用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为记录每个文件夹里数字的个数
};