剑指offer：包含min函数的栈 python实现

题目

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

解题思路

暴力做法：

1. 遍历一遍，找最小的数，然后返回。

# -*- coding:utf-8 -*-
class Solution:
    def __init__(self):
        self.mys=[]
    def push(self, node):
        # write code here
        self.mys.append(node)
    def pop(self):
        # write code here
        self.mys.pop()
    def top(self):
        # write code here
        self.mys.pop(1)
    def min(self):
        # write code here
        if len(self.mys)==0:
            return None
        min_num = self.mys[0]
        for i in self.mys:
            if i < min_num:
                min_num = i
        return min_num
            

自己的思路太暴力，晚上回去看看书。

做了几家笔试，发现同样的思路c++能过，但python却超时了。
只能开始转战c++啦！

剑指offer解题思路：

1. 建立两个栈，数据栈用来存放每次存的数据；辅助栈用来存当前最小的数据，同时能找到次小的数据。
2. 存入辅助栈的时候要判断该栈是否为空，为空则存放该值；
3. 不为空，则判断该值是否小于辅助栈的栈顶元素（即最小值），小于则存放该值；
4. 不小于则存放辅助栈的栈顶元素。

class Solution {
public:
    void push(int value) {
        m_data.push(value);
        if(m_min.size()==0 || value<m_min.top()){
            m_min.push(value);
        }
        else{
            m_min.push(m_min.top());
        }
    }
    void pop() {
        if(m_data.size()>0 && m_min.size()>0){
            m_data.pop();
            m_min.pop();
        }          
    }
    int top() {
        if(m_data.size()>0 && m_min.size()>0){
            return m_data.top();
        }
    }
    int min() {
        if(m_data.size()>0 && m_min.size()>0){
            return m_min.top();
        }
    }
private:
    stack<int> m_data;
    stack<int> m_min;      
};