劍指offer-包含min函數的棧（python和c++）

題目描述

定義棧的數據結構，請在該類型中實現一個能夠得到棧中所含最小元素的min函數（時間複雜度應爲O（1））。
注意：保證測試中不會當棧爲空的時候，對棧調用pop()或者min()或者top()方法。

思路：
維護兩個棧A,B，一個A用來存棧中所有元素，一個B是存棧中最小的元素。

push函數中，對於A來說，直接使用append方法添加到末尾，對於B需要判斷棧中最後一個元素是否大於新元素，從而決定是否入棧B。

pop函數，先判斷A,B兩棧末尾是否相等，直接返回末尾元素即可。

top函數直接返回棧A 最後一個元素。

min函數直接返回棧B最後一個元素即爲最小元素。

代碼很簡單，如下。

python

# -*- coding:utf-8 -*-
class Solution:
    def __init__(self):
        self.stack = []
        self.min_stack = []
    def push(self, node):
        # write code here
        self.stack.append(node)
        if not self.min_stack or node <= self.min_stack[-1]:
            self.min_stack.append(node)
    def pop(self):
        # write code here
        if self.stack[-1] == self.min_stack[-1]:
            self.min_stack.pop()
        self.stack.pop()
    def top(self):
        # write code here
        return self.stack[-1]
    def min(self):
        # write code here
        return self.min_stack[-1]

c++

class Solution {
public:
    void push(int value) {
        stk.push(value);
        if(stk_min.empty())
            stk_min.push(value);
        else if(value <= stk_min.top())
            stk_min.push(value);
    }
    void pop() {
        //stk.pop();
        if(stk.top() == stk_min.top())
            stk_min.pop();
        stk.pop();
    }
    int top() {
        return stk.top();
    }
    int min() {
        return stk_min.top();
    }
private:
    stack<int> stk;
    stack<int> stk_min;
};