-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathMinStack.py
101 lines (78 loc) · 1.83 KB
/
MinStack.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
class MinStack(object):
def __init__(self):
"""
initialize your data structure here.
min = 4 2
4 2 6 1
0 -2 -1
"""
self.stack = []
self.min = []
def push(self, x):
"""
:type x: int
:rtype: nothing
"""
self.stack.append(x)
if len(self.min) == 0 or x <= self.min[-1]:
self.min.append(x)
def pop(self):
"""
:rtype: nothing
"""
top = self.stack[-1]
self.stack.pop()
if top == self.min[-1]:
self.min.pop()
def top(self):
"""
:rtype: int
"""
if len(self.stack) == 0:
return None
return self.stack[-1]
def getMin(self):
"""
:rtype: int
"""
return self.min[-1]
class MinStack(object):
def __init__(self):
"""
initialize your data structure here.
"""
self.min = 0
self.stack = []
def push(self, x):
"""
:type x: int
:rtype: nothing
"""
if len(self.stack) == 0:
self.stack.append(0)
self.min = x
else:
self.stack.append(x - self.min)
self.min = min(self.min, x)
def pop(self):
"""
:rtype: nothing
"""
if len(self.stack) == 0:
return
pop = self.stack.pop()
if pop < 0:
self.min -= pop
def top(self):
"""
:rtype: int
"""
if len(self.stack) == 0:
return None
top = self.stack[-1]
return self.min + (top if top > 0 else 0)
def getMin(self):
"""
:rtype: int
"""
return self.min