python实现时间o(1)的最小栈的实例代码
这是毕业校招二面时遇到的手写编程题,当时刚刚开始学习python,整个栈写下来也是费了不少时间。毕竟语言只是工具,只要想清楚实现,使用任何语言都能快速的写出来。
何为最小栈?栈最基础的操作是压栈(push)和退栈(pop),现在需要增加一个返回栈内最小值的函数(get_min),要求get_min函数的时间复杂度为o(1)。python的栈肯定是使用list实现,只要将list的append和pop封装到stack类中,即实现了压栈和退栈。如果不考虑时间复杂度,我们第一反应一定是min(),min()可以在不开辟新空间的情况下o(n)的返回栈内最小值。但是如果栈内元素很多,并且get_min方法需要频繁调用时,min高耗时的缺点就被放大,那么理想的方法就是空间换时间来降低时间复杂度。
我们的栈内存在stack_list和min_list,min_list负责存储栈内元素中最小值组成的栈,当新压栈的元素小于等于栈内最小的元素时,将新元素压入min_list。如果退栈的元素等于栈内最小的元素,那么也要将min_list退栈。举例子,我们依次压栈3,2,4,1
初始化
stack_list = [] min_list = []
3压栈
stack_list = [3] min_list = [3]
2压栈
stack_list = [3, 2] min_list = [3, 2]
4压栈
stack_list = [3, 2, 4] min_list = [3, 2]
1压栈
stack_list = [3, 2, 4, 1] min_list = [3, 2, 1]
get_min只需要返回min_list中最后一个元素,以下是python代码的完整实现
#!/usr/bin/python # -*- coding: utf-8 -*- class stack(object): stack_list = [] min_list = [] min = None def push(self, x): if not self.stack_list: self.min = x self.min_list.append(self.min) self.stack_list.append(x) return self.stack_list.append(x) if self.min >= x: self.min = x self.min_list.append(self.min) return def pop(self): pop_result = None if self.stack_list: pop_result = self.stack_list[-1] if self.stack_list.pop() == self.min: self.min_list.pop() if self.min_list: self.min = self.min_list[-1] else: self.min = None return pop_result else: self.min = None return pop_result def print_stack(self): print "stack---->", self.stack_list return def get_min(self): return self.min
以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持脚本之家。
相关文章
Python Pickle 实现在同一个文件中序列化多个对象
今天小编就为大家分享一篇Python Pickle 实现在同一个文件中序列化多个对象,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧2019-12-12opencv中颜色空间转换函数cv2.cvtColor()使用
本文主要介绍了opencv中颜色空间转换函数cv2.cvtColor()使用,文中通过示例代码介绍的非常详细,具有一定的参考价值,感兴趣的小伙伴们可以参考一下2022-05-05使用python tkinter开发一个爬取B站直播弹幕工具的实现代码
这篇文章主要介绍了使用python tkinter开发一个爬取B站直播弹幕的工具,本文通过实例代码给大家介绍的非常详细,对大家的学习或工作具有一定的参考借鉴价值,需要的朋友可以参考下2021-02-02pyinstaller打包python3.6和PyQt5中各种错误的解决方案汇总
pyinstaller是打包python很方便的一个套件,我们可以很轻易地使用他,下面这篇文章主要给大家介绍了关于pyinstaller打包python3.6和PyQt5中各种错误解决的相关资料,需要的朋友可以参考下2022-08-08
最新评论