(剑指Offer-牛客网)包含min函数的栈

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

借鉴了他人思路。自己分析不够,以为需要排序,其实栈的“先进后出”规则决定了:如果后压入的元素都不影响当前最小值,那么这些元素在出栈时也不影响最小值。因此只需要用stack模拟常规的栈,用min_record记录在压栈过程中会影响到最小值的那些元素即可。

注意:1. end() 指向的是尾后元素,取尾元素需要减一; 2. 在pop中先进行元素比较再执行出栈操作

class Solution {
public:
    vector stack;
    vector min_record;
    
    void push(int value) {
        stack.push_back(value);
        if(min_record.empty()||value <= *(min_record.cend()-1))
            min_record.push_back(value);
    }
    void pop() {
        if(!stack.empty())
            if(top()==*(min_record.cend()-1))
                min_record.erase(min_record.end()-1);
            stack.erase(stack.end()-1);
    }
    int top() {
        if(!stack.empty())
            return *(stack.cend()-1);
    }
    int min() {
        if(!min_record.empty())
            return *(min_record.cend()-1);
    }
};

 

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