LeetCode-155. 最小栈

设计一个支持 push，pop，top 操作，并能在常数时间内检索到最小元素的栈。

push(x) -- 将元素 x 推入栈中。
pop() -- 删除栈顶的元素。
top() -- 获取栈顶元素。
getMin() -- 检索栈中的最小元素。
示例:

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

 

如果自己实现数据结构，代码要更复杂一些，这里偷了个懒，用了stl当中的stack容器

m_stack正常实现栈的压入和弹出。

m_assist辅助栈，用于存取最小值，如果与m_stack栈顶元素相等则弹出，

实现一个递增的单调栈，保证辅助栈栈顶元素一直是最小，返回的时候返回辅助栈的栈顶即可。

#include <iostream>
#include <stdlib.h>
#include <stack>

/**
* Your MinStack object will be instantiated and called as such:
* MinStack* obj = new MinStack();
* obj->push(x);
* obj->pop();
* int param_3 = obj->top();
* int param_4 = obj->getMin();
*/

using namespace std;

class MinStack {
public:
	/** initialize your data structure here. */
	MinStack() {

	}

	void push(int x) {
		m_stack.push(x);
		if (m_assist.empty()) {
			m_assist.push(x);
		}
		else if (m_assist.top() >= x) {
			m_assist.push(x);
		}
	}

	void pop() {
		if (m_stack.top() == m_assist.top()) {
			m_assist.pop();
		}
		m_stack.pop();

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

	int getMin() {
		return m_assist.top();
	}

private:
	stack<int> m_stack;   
	stack<int> m_assist;
};

int main() {
	MinStack minStack = MinStack();
	minStack.push(-2);
	minStack.push(0);
	minStack.push(-3);
	cout << minStack.getMin() << endl;;   //--> 返回 - 3.
	minStack.pop();
	cout << minStack.top() << endl;      //--> 返回 0.
	
	cout << minStack.getMin() << endl;;


	system("pause");
	return 0;
}