设计包含min函数的栈

设计包含min函数的栈。
定义栈的数据结构,要求添加一个min函数,能够得到栈的最小元素。
要求函数min、push以及pop的时间复杂度都是O(1)。

栈中元素设两个域,一个记录元素的值,一个记录栈的最小值。

class Stack(object):
    def __init__(self):
        self.list=[]
        self.top=-1
    def push(self,x):
        if self.top>=0:
            if x<self.list[-1][1]:
                self.list.append((x,x))
            else:
                self.list.append((x,self.list[-1][1]))
        else:
            self.list.append((x,x))
        self.top+=1
    def pop(self):
        if self.top>=0:
            self.top-=1
            return self.list.pop()[0]
        else:
            raise IndexError,'pop from empty stack'
    def min(self):
        if self.top>=0:
            return self.list[self.top][1]
        else:
            raise IndexError,'stack is empty'

  

posted @ 2014-06-17 22:50  再见紫罗兰  阅读(720)  评论(0编辑  收藏  举报