【剑指offer】面试题30 包含min的栈

有关剑指offer题目的解析:剑指offer 题目整理

题目要求

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

解题分析

看到这个题目,我们就可以有一种想法就是,设置一个变量存放栈中最小的元素,比较每次压入栈中的元素,不断更新。 这是一个办法,可以帮助我们找到栈中最小的元素,但是如果弹出这个最小元素之后,我们还想获得剩下栈中的最小元素怎么办??这时候,设置变量的方法显然是不够的,因为我们还需要知道次小元素,所以为了方便操作,我们设置一个辅助栈 // 辅助栈 每次把元素入数据栈的同时把当前最小元素压入辅助栈什么意思呢?见下图,示例 3-4-2-1
【剑指offer】面试题30 包含min的栈_第1张图片
这样做的好处是,实际上我们保存了当前栈中的最小元素,不论后续再做弹出,压入的操作都不影响,因为每次我们都会把结果存放在辅助栈中,只要读取辅助栈的栈顶就是我们想要的最小元素。

注意的点:当前最小元素的判断,辅助栈的栈顶和输入元素进行比较即可。

主要代码c++

class Solution {
public:
    stack<int> data_stack, min_stack;
    void push(int value) {
        data_stack.push(value);
        if(min_stack.size()==0 || value < min_stack.top()) // 判断当前最小值
            min_stack.push(value);
        else
            min_stack.push(min_stack.top());
    }
    void pop() {
        if(data_stack.top()==min_stack.top())
            return data_stack.pop(); // 结合top
        return min_stack.pop(); // 结合min
    }
    int top() {
       return data_stack.top();
    }
    int min() {
       return min_stack.top();
    } 
};

你可能感兴趣的:(剑指offer)