剑指offer面试题21：包含min函数的栈

最新推荐文章于 2021-08-29 11:12:07 发布

原创最新推荐文章于 2021-08-29 11:12:07 发布 · 247 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#面试题

剑指offer 专栏收录该内容

33 篇文章

订阅专栏

本文介绍了一种特殊的数据结构——带有最小值查询功能的栈。通过维护两个栈，一个用于存储常规数据，另一个用于跟踪最小值历史记录，实现了在O(1)时间内完成push、pop和min操作。

题目：定义栈的数据结构，请在该类型中实现一个能够得到栈的最小元素的min函数。在该栈中，调用min、push及pop的时间复杂度都是O(1)。

template <typename T> void StackWithMin<T>::push(const T &value) {
  m_data.push(value);

  if (m_min.size() == 0 || value < m_min.top())
    m_min.push(value);
  else
    m_min.push(m_min.top());
}

template <typename T> void StackWithMin<T>::pop() {
  assert(m_data.size() > 0 && m_min.size() > 0);

  m_data.pop();
  m_min.pop();
}

template <typename T> const T &StackWithMin<T>::min() const {
  assert(m_data.size() > 0 && m_min.size() > 0);

  return m_min.top();
}