题目描述
定义栈的数据结构,请在该类型中实现一个能够得到栈中所含最小元素的min函数(时间复杂度应为O(1))。
思路
设定一个数据栈(真实数据入栈)和一个辅助栈(每次新元素入数据栈时,将此时数据站的最小元素压入辅助栈),数据栈弹出元素时辅助栈同时也弹出栈顶元素. 任意时刻辅助栈栈顶元素即为此时数据栈中的最小元素
代码
#include <assert.h>
class Solution {
public:
/*
void push(int value) {
//元素入数据栈
m_data.push(value);
//如辅助栈为空 或 辅助栈顶元素>欲入栈元素
if(m_min.size() == 0 || m_min.top() > value)
m_min.push(value);
//否则,将辅助栈顶元素再次入栈
else
m_min.push(m_min.top());
}
void pop() {
//两栈均不空时,分别pop
assert(m_data.size() > 0 && m_min.size() > 0);
m_data.pop();
m_min.pop();
}
int top() {
return m_data.top();
}
int min() {
assert(m_data.size() > 0 && m_min.size() > 0);
return m_min.top();
}
*/
void push(int value) {
//元素入数据栈
m_data.push(value);
//如辅助栈为空 或 辅助栈顶元素>欲入栈元素
if(m_min.size() == 0 || m_min.top() > value)
m_min.push(value);
//否则,将辅助栈顶元素再次入栈
else
m_min.push(m_min.top());
}
void pop() {
//两栈均不空时,分别pop
//assert(m_data.size() > 0 && m_min.size() > 0);
m_data.pop();
m_min.pop();
}
int top() {
return m_data.top();
}
int min() {
//assert(m_data.size() > 0 && m_min.size() > 0);
return m_min.top();
}
private:
//数据栈
stack<int> m_data;
//辅助栈
stack<int> m_min;
};