O(1)时间复杂度实现入栈、出栈、获得栈中最小元素、获得栈中最大元素

最新推荐文章于 2024-08-09 21:26:42 发布

happy309best

最新推荐文章于 2024-08-09 21:26:42 发布

阅读量2.3k

点赞数 2

分类专栏： algorithm design 文章标签：面试题算法设计栈

本文链接：https://blog.csdn.net/happy309best/article/details/47725935

版权

该博客讨论了一道面试题，要求设计一个数据结构，使得在O(1)时间复杂度内完成入栈、出栈操作，并能动态获取栈中最小和最大元素。解决方案是利用辅助栈来存储最小和最大值，确保所有操作高效进行。

摘要由CSDN通过智能技术生成

题目要求：定义栈的数据结构，添加min()、max()函数（动态获取当前状态栈中的最小元素、最大元素），要求push()、pop()、min()、max()的时间复杂度都是O(1)。

思路解析：根据栈的后进先出特性，增加辅助栈，来存储当前状态下数据栈中的最小、最大元素。

代码：（python实现）

#coding=utf-8

class Stack(object):
    
    def __init__(self):
        self.data = []
        self.minValue = []
        self.maxValue = []
    
    def push(self, data):
        self.data.append(data)
        if len(self.minValue) == 0:
            self.minValue.append(data)
        else:
            if data <= self.minValue[-1]:
                self.minValue.append(data)
        if len(self.maxValue) == 0:
            self.maxValue.append(data)