程序员面试题精选100题(02)-设计包含min函数的栈—python实现

6 篇文章 0 订阅
5 篇文章 0 订阅

本作品采用知识共享署名-非商业性使用-相同方式共享 3.0 Unported许可协议进行许可。允许非商业转载,但应注明作者及出处。


作者:liuyuan_jq

2011-07-13




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

该思路存在一个重要的问题:如果当前最小元素被pop出去,如何才能得到下一个最小元素?

因此仅仅只添加一个成员变量存放最小元素(或最小元素的位置)是不够的。我们需要一个辅助栈。每次push一个新元素的时候,同时将最小元素(或最小元素的位置。考虑到栈元素的类型可能是复杂的数据结构,用最小元素的位置将能减少空间消耗)push到辅助栈中;每次pop一个元素出栈的时候,同时pop辅助栈。


#!/usr/bin/env python
# -*- coding:utf-8 -*-

"""
题目:定义栈的数据结构,要求添加一个min函数,能够得到栈的最小元素。要求函数min、push以及pop的时间复杂度都是O(1)。
分析:这是去年google的一道面试题。
"""

class MinStack(object):
    def __init__(self):
        self.data     = []
        self.minIndex = []
    
    def top(self):
        return self.data[-1]
    
    def push(self, value):
        self.data.append(value)
        if not self.minIndex:
            self.minIndex.append(0)
        else:
            if value < self.data[self.minIndex[-1]]:
                self.minIndex.append(len(self.data) - 1)
            else:
                self.minIndex.append(self.minIndex[-1])
        self.dump()
        
    def pop(self):
        value = self.data.pop()
        self.minIndex.pop()
        self.dump()
        return value
    
    def min(self):
        return self.data[self.minIndex[-1]]
    
    def dump(self):
        print str(self.data).ljust(20), str(self.minIndex).ljust(20), str(self.min()).ljust(20)

    
if __name__ == '__main__':
    stack = MinStack()
    print "Stack".ljust(20), "MinStack".ljust(20), "Min".ljust(20)
    stack.push(3)
    stack.push(4)
    stack.push(2)
    stack.push(1)
    stack.pop()
    stack.pop()
    stack.push(5)
    stack.push(0)


结果

Stack                MinStack             Min                 
[3]                  [0]                  3                   
[3, 4]               [0, 0]               3                   
[3, 4, 2]            [0, 0, 2]            2                   
[3, 4, 2, 1]         [0, 0, 2, 3]         1                   
[3, 4, 2]            [0, 0, 2]            2                   
[3, 4]               [0, 0]               3                   
[3, 4, 5]            [0, 0, 0]            3                   
[3, 4, 5, 0]         [0, 0, 0, 3]         0                   



评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值