《剑指offer》-包含min函数的栈

原创于 2016-05-11 20:40:24 发布 · 410 阅读

0 ·

CC 4.0 BY-SA版权

编程刷题专栏收录该内容

47 篇文章

订阅专栏

本文介绍了一种特殊栈的设计方案，通过维护一个辅助栈来高效获取栈内最小元素。当新元素加入时，会与辅助栈顶元素比较，确保辅助栈顶始终是最小值。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目描述

定义栈的数据结构，请在该类型中实现一个能够得到栈最小元素的min函数。

解法：借助一个辅助栈，入栈的同时，判断入栈值是否大于辅助栈的头指针的值，大于则入栈，小于则入自己的头指针值。始终保持头指正值是入栈的最小值。

#include <assert.h>
class Solution {
public:
    stack<int> m_data, m_min;
    void push(int 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());
    }
    void pop() {
        assert(m_data.size() >0 && m_min.size() > 0);
        m_data.pop();
        m_min.pop();
    }
    int top() {
        assert(m_data.size() >0 && m_min.size() > 0);
        return m_data.top();
    }
    int min() {
        assert(m_data.size() >0 && m_min.size() > 0);
        return m_min.top();
        
    }
};