剑指offer:包含min函数的栈

it2022-05-05  154

题目描述 定义栈的数据结构,请在该类型中实现一个能够得到栈中所含最小元素的min函数(时间复杂度应为O(1))。

思路: 因为时间复杂度要在O(1),一般的遍历的方法肯定不可行。于是我们想到可以利用一个辅助的栈min来存放最小值。 入栈时:先进入data栈,与辅助栈min栈顶比较大小,若小的话就进入min栈,若不小于min的栈顶,则min的栈顶入min栈. 出栈时:data栈和min栈都要出栈 实现:

class Solution { public: stack <int> data;//这个栈用来存储所有的 stack <int> min1;//这个栈用来存储最小的值 void push(int value) { data.push(value); if(min1.empty()||min1.top()>value) { min1.push(value); } else{ min1.push(min1.top()); } } void pop() { data.pop(); min1.pop(); } int top() { return data.top(); } int min() { return min1.top(); } };

最新回复(0)