这道题目的精妙之处在于题目已经给出了答案。
乍看题目描述,只需在成员里增添一个变量min,每次push的时候维护这个min值就可以了。
请原谅我这一位算法渣居然没有想pop的时候怎么办就已经提交了,还信心满满。
WA给出了错误信息。我瞬间就明白自己错在哪。
push维护了min值,pop呢?
开始我想看看pop出去的值是不是等于min,等于就。。。
哎?我总不能再找第二大的值吧,何况这是栈,根本就不能扫描一遍(或者说扫描很麻烦)
看着正确答案的输入输出,再联想题目
Min Stack!!
让min作为一个stack不就可以了么?
每次让push进minstack的元素都与栈顶元素比较,小的就插入,大的就继续插入原栈顶元素。
这样一来问题就解决了。
class MinStack {
public:
void push(int x) {
int len = sto.size();
if (len == 0) {
min.push(x);
}
else if (min.top() > x) {
min.push(x);
}
else min.push(min.top());
sto.push(x);
}
void pop() {
sto.pop();
min.pop();
}
int top() {
return sto.top();
}
int getMin() {
return min.top();
}
public:
stack<int> sto;
stack<int> min;
};
这样的minstack还有一个性质,它是递减的。
很高兴能够和solution的思路一样,而且我发现我之前很多做不出来的题目现在也有思路了。