设计getMin功能的栈

最新推荐文章于 2019-10-29 19:24:47 发布

原创最新推荐文章于 2019-10-29 19:24:47 发布 · 192 阅读

0 ·

CC 4.0 BY-SA版权

程序员代码面试指南专栏收录该内容

3 篇文章

订阅专栏

该博客介绍如何设计一个特殊的栈，除了基本的栈操作外，还额外提供了获取栈中最小元素的功能。通过详细讨论输入和输出描述，以及提供具体的代码实现，作者展示了如何在不增加额外空间复杂度的情况下，高效地实现这一功能。

题目描述

实现一个特殊功能的栈，在实现栈的基本功能的基础上，再实现返回栈中最小元素的操作。

输入描述

第一行输入一个整数N，表示对栈进行的操作总数。

下面N行每行输入一个字符串S，表示操作的种类。

如果S为"push"，则后面还有一个整数X表示向栈里压入整数X。

如果S为"pop"，则表示弹出栈顶操作。

如果S为"getMin"，则表示询问当前栈中的最小元素是多少。

输出描述

对于每个getMin操作，输出一行表示当前栈中的最小元素是多少。

代码实现

N = int(input().strip())
stack = []
min_express = []
for i in range(N):
    temp = input().split()
    flag = temp[0]
    if flag == 'push':
        num = int(temp[1])
        stack.append(num)
        if len(min_express) == 0:
            min_express.append(num)
        else:
            if num < min_express[-1]:
                min_express.append(num)
            else:
                min_express.append(min_express[-1])
        continue
    if flag == 'pop':
        stack.pop()
        min_express.pop()
        continue
    if flag == 'getMin':
        print(min_express[-1])
        continue

从评论中看到一个规整的代码实现，在下面展示：

import sys 
 
class GetMinStack(object):
    def __init__(self):
        self.stackData=[]
        self.stackMin=[]
    def push(self,num):
        self.stackData.append(num)
        if len(self.stackMin)==0 or num<=self.getMin():
            self.stackMin.append(num)
        else:
            self.stackMin.append(self.getMin())
    def pop(self):
        if len(self.stackData)==0:
            raise Exception("stack is empty")
        self.stackMin.pop()
        return self.stackData.pop()
    def getMin(self):
        if len(self.stackMin)==0:
            raise Exception("stack is empty")
        return self.stackMin[-1]
 
N = int(sys.stdin.readline().strip())
getMinStack = GetMinStack()
for i in range(N):
    line = sys.stdin.readline().strip()
    words = line.split()
    if words[0] == "push":
        getMinStack.push(int(words[1]))
    elif words[0] == "pop":
        getMinStack.pop()
    else:
        print(getMinStack.getMin())