包含min函数的栈|剑指offer

来源:互联网 发布:ubuntu查看空间大小 编辑:程序博客网 时间:2024/04/29 08:45

题目描述

定义栈的数据结构,请在该类型中实现一个能够得到栈最小元素的min函数
因为栈结构先进后出的,如果我们只使用一个变量业保存最小值,是不可行的。我们可以构造一个辅助函数来记录每个位置上对应的最小值。剑指offer里面的原题要求实现的每个函数操作的时间都是O(1),所以每次操作的时候就要对最小值辅助结构进行更新,这样当获得最小值的时候就可以是O(1)复杂度。


class Solution {private:    std::stack<int> val;    std::stack<int> minVal;public:    void push(int value) {        val.push(value);        if(minVal.empty() || (value < minVal.top())){            minVal.push(value);        }           else{               minVal.push(minVal.top());           }       }               void pop() {        val.pop();        minVal.pop();            }    int top() {        return val.top();            }    int min() {        return minVal.top();    }};

但是这里面并没有做一个处理,也就是,当我们的栈为空没有元素时,对其进行弹出和返回栈顶元素是不行的,最好在里面再加上一句

assert(val.size()>0 && minVal.size() > 0);

牛客网的编译环境里面没有包含assert的头文件,而且不能够自己添加,所以就没有做错误处理了。

另外一种方法是使用pair来减少一个stack的使用,少了一个类型。

#define MIN(a,b) (a >b ?b:a)class Solution {    stack<pair<int, int>> val;public:    void push(int value) {        val.push(pair<int, int>(value, val.empty()?value:MIN(value, min())));      }    void pop() {        val.pop();    }    int top() {        return val.top().first;           }    int min() {        return val.top().second;            }};



0 0
原创粉丝点击