包含min函数的栈

来源:互联网 发布:淘宝怎么设置客服号 编辑:程序博客网 时间:2024/06/07 10:50

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

解法1

双栈或pair,一个存储数据,一个用来存储当前栈中最小值

class Solution {public:    void push(int value) {        haha.push(pair<int, int>(value, haha.empty()? value : (value < min() ? value : min())));    }    void pop() {        haha.pop();     }    int top() {        return haha.top().first;    }    int min() {        return haha.top().second;    }private:    stack<pair<int, int>> haha;};

解法二

不借用辅助栈,只用一个变量minValue,空间复杂度O(1), 原栈只存value与minValue的差值,注意push和pop时的操作

push:第一次入0,赋值给minValue,之后若发现比minValue小,则先入栈(这里就是旧minValue与新minValue的差)再更新minValue
pop : 如果发现栈顶为负,说明要将minValue回退到旧的
top : 如果栈顶为正,返回minValue+top(),否则说明是应是当前最小值,返回minValue。

如3、4、1 开始0入栈,minvalue为3,之后4-3=1入栈,1-3=-2入栈,minvalue为1,此时若要出栈,发现-2为负,要恢复minValue为-2(1)还没入时的状态,即1-(-2)=3,若要返回栈顶,则需直接返回minValue为1。

class Solution {public:    void push(int value) {        if(haha.empty()){            haha.push(0);            minValue = value;        }else{            haha.push(value - minValue);            if(value < minValue)    minValue = value;        }    }    void pop() {        if(haha.top() < 0)  minValue -= haha.top();        haha.pop();    }    int top() {        if(haha.top() > 0)  return haha.top() + minValue;        return minValue;    }    int min() {        return minValue;    }private:    stack<int> haha;    int minValue;};
0 0
原创粉丝点击