剑指offer 包含min函数的栈

Posted on |

包含min函数的栈

题目

Leetcode 155.最小栈
定义栈的数据结构，请在该类型中实现一个能够得到栈中所含最小元素的min函数（时间复杂度应为O（1））

方法

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
# -*- coding:utf-8 -*-
class Solution:
    def __init__(self):
        self.stack=[]
    def push(self, node):
        # write code here
        if self.stack:
            self.stack.append((node,min(node,self.stack[-1][1])))
        else:
            self.stack.append((node,node))

    def pop(self):
        # write code here
        if self.stack:
            self.stack.pop()
    def top(self):
        # write code here
        if self.stack:
            return self.stack[-1][0]
    def min(self):
        # write code here
        if self.stack:
            return self.stack[-1][1]