题目描述:
定义栈的数据结构,请在该类型中实现一个能够得到栈中所含最小元素的min函数(时间复杂度应为O(1))。
解题思路:
相当与在保留原栈的同时,去维护一个最小栈。利用一个辅助栈来完成。对于每个新输入的数,在压入原栈的同时,需要去判断是否为最小值,若为最小指,则压入辅助栈,否则将当前最小值压入辅助栈。这样就相当于两个栈的大小是相等的,每次做push和pop操作时,二者需要同步。最后的min函数只需要取辅助栈的top就可以。
代码:
class Solution { public: stack<int>cur, min_stack; void push(int value) { cur.push(value); if(min_stack.empty()) min_stack.push(value); else { if(value < min_stack.top()) min_stack.push(value); else min_stack.push(min_stack.top()); } } void pop() { if(!cur.empty()) { cur.pop(); min_stack.pop(); } } int top() { if(!cur.empty()) return cur.top(); return 0; } int min() { if(!min_stack.empty()) { return min_stack.top(); } return 0; } };