js 包含min函数的栈_包含min函数的栈

包含min函数的栈

题目描述

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

思路, 若使用标记用来标记最小元素的位置, 若弹出的是最小元素, 还需要次最小元素的标记位, 若再弹出的是次最小元素, 又需要次次最小元素, 依次类推, 则不能确定需要多少个标记位

使用两个栈, 一个用来当数据栈, 一个用来装最小值的栈, 两个栈的压入与弹出操作同时进行, 栈的最小值永远位于最小值栈的栈顶

当压入元素小于最小值栈时, 数据栈与最小值栈同时压入相同元素

当压入元素大于等于的于最小值栈的栈顶时, 数据栈正常压入, 最小值栈压入最小值栈的栈顶元素

这样, 两个栈弹出操作同步执行, 当弹出元素不是最小值时, 由于之前压入该值时, 最小栈压入是当时栈顶元素, 因此, 弹出后最小栈的栈顶还是最小值

class Solution {

public:

stack m_data;

stack m_min;

void push(int value) {

m_data.push(value);

if (m_min.empty() || value < m_min.top()) {

m_min.push(value);

}

else {

m_min.push(m_min.top());

}

}

void pop() {

m_data.pop();

m_min.pop();

}

int top() {

return m_data.top();

}

int min() {

return m_min.top();

}

};