设计getMin功能的栈

题目描述

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

输入描述

第一行输入一个整数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()
        continu