琦琦的彩虹糖
网友问我的一道题 不知套题目名字 就叫:琦琦的彩虹糖吧)
题目难度不高 主要考察stack的应用
摘录的原因是储存栈中最大值的方法很好,压栈的时候push value的同时还push max 这样就可以一直保持stack top的值是最大值 这个思路非常好 (之前在leetcode上有看过类似题解 之后没用过就忘了 这次赶紧摘录)!
网友的题解(帮网友改过代码之后 只能过80%测试点 实在是不知道怎么能优化 据说最后一个测试点是超时了),要是有网友看到这题 有压缩时间的改进思路 希望能留下你宝贵的意见呀!!
#include <iostream>
#include <stack>
using namespace std;
int main()
{
//输入m
int m;
cin>>m;
int i=0;//记录操作次数
stack<int> sta;//放入糖的栈
int max=-1;//记录糖度最高的值
//vector<int> res;//保存要输出的值
int x;
int value;//美味度
while (i<m){
cin>>x;//输入要进行的操作
if (x==1){
cin>>value;
if (value>max){//如果新放入的糖比最大的还美味,则改变max的值
max=value;
}
sta.push(value);
sta.push(max);
cout<<max<<endl;
}else {
if (sta.size()==0){
i++;
cout<<-1<<endl;
continue;
}
if (sta.size()==2){//意味着这一次拿出去以后就没有糖了
sta.pop();
sta.pop();
max=-1;
cout<<-1<<endl;
}else {
sta.pop();
sta.pop();
cout<<sta.top()<<endl;
}
}
i++;
}
return 0;
}