定义栈的数据结构,请在该类型中实现一个能够得到栈中所含最小元素的min函数(时间复杂度应为O(1))。
class Solution { public: stack<int> s1,s2; void push(int value) { s1.push(value); if(s2.empty()){ s2.push(value); } else{ if(value<=s2.top()){ s2.push(value); } } } void pop() { if(s1.top()==s2.top()){ s2.pop(); } s1.pop(); } int top() { return s1.top(); } int min() { return s2.top(); } };
思路:用一个辅助栈记录入栈元素从大到小,如果后面压入的元素比辅助栈顶元素大,就不用压入它(等于还是要,否则pop了会出错)。考虑635入栈,第一次s1:6,s2:6,min:6。第二次s1:6,3,s2:6,3,min:3,第三次s1:6,3,5,s2:6,3,min:3。pop时要注意如果pop的是最小值,s2也要pop