Leetcode#155 Min Stack

原题地址

 

非常经典的题目了。双栈法，一个普通栈，一个最小值栈

入栈时：除了入普通栈之外，如果当前待入栈元素小于等于最小值栈顶元素，那么同时入最小值栈

出栈时：除了出普通栈之外，如果当前待出栈元素小于等于最小值栈顶元素，那么同时出最小值栈

查看栈顶元素：返回普通栈顶元素

查看最小元素：返回最小值栈顶元素

 

代码：

stack<int> st;
stack<int> minSt;

void push(int x) {
  st.push(x);
  if (minSt.empty() || x <= minSt.top())
    minSt.push(x);
}

void pop() {
  if (st.empty()) return;
  if (st.top() == minSt.top())
    minSt.pop();
  st.pop();
}

int top() {
  if (st.empty()) return -1;
  else return st.top();
}

int getMin() {
  if (st.empty()) return -1;
  else return minSt.top();
}