定义栈的数据结构,请在该类型中实现一个能够得到栈中所含最小元素的min函数(时间复杂度应为O(1))。
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33
| class Solution { public: stack<int> data, mins; void push(int value) { data.push(value); if (mins.size() > 0) { const int old_min = mins.top(); if (old_min < value) { mins.push(old_min); } else { mins.push(value); } } else { mins.push(value); } } void pop() { data.pop(); mins.pop(); } int top() { return data.top(); } int min() { return mins.top(); } };
|