0
点赞
收藏
分享

微信扫一扫

面试题 03.02. 栈的最小值

幸福的无所谓 2022-04-18 阅读 49
leetcode

面试题 03.02. 栈的最小值
请设计一个栈,除了常规栈支持的pop与push函数以外,还支持min函数,该函数返回栈元素中的最小值。执行push、pop和min操作的时间复杂度必须为O(1)。

示例:

MinStack minStack = new MinStack();
minStack.push(-2);
minStack.push(0);
minStack.push(-3);
minStack.getMin(); --> 返回 -3.
minStack.pop();
minStack.top(); --> 返回 0.
minStack.getMin(); --> 返回 -2.

代码

class MinStack {
	stack<int> s;
	stack<int> min;
	int minV;
	bool isInit = false;
public:
	/** initialize your data structure here. */
	MinStack() {
	}

	void push(int x) {
		s.push(x);
		if (!isInit)
		{
			isInit = true;
			minV = x;
		}
		if (x<minV)
		{
			minV = x;
		}
		min.push(minV);
	}

	void pop() {
		s.pop();
		min.pop();
		//这个判断很重要 删除最后一个元素 会报错
		if (min.empty())
		{	
			isInit=false; 
		}
		else
		{
			minV = getMin();
		}
	}

	int top() {
		return s.top();
	}

	int getMin() {
		return min.top();
	}
};
举报

相关推荐

0 条评论