This approach utilizes two stacks. One stack will hold the main elements, while the other stack will keep track of the minimum elements. When you push a value, it's added to the main stack, and if it's less than or equal to the current minimum, it's also pushed onto the min stack. During a pop operation, if the value being popped is the current minimum, it's also popped from the min stack.
Time Complexity: O(1) for each operation.
Space Complexity: O(n), where n is the number of elements in the stack.
1class MinStack:
2 def __init__(self):
3 self.stack = []
4 self.min_stack = []
5
6 def push(self, val: int) -> None:
7 self.stack.append(val)
8 if not self.min_stack or val <= self.min_stack[-1]:
9 self.min_stack.append(val)
10
11 def pop(self) -> None:
12 if self.stack.pop() == self.min_stack[-1]:
13 self.min_stack.pop()
14
15 def top(self) -> int:
16 return self.stack[-1]
17
18 def getMin(self) -> int:
19 return self.min_stack[-1]
20The Python implementation uses two lists (stacks). The main stack operates as a typical stack, while the min_stack keeps track of the current minimums. Push and pop operations involve conditionally modifying min_stack based on whether the pushed or popped values affect the current minimum.
This approach reduces space usage by storing a pair (value, minimum so far) in a single stack. Each element maintains the current minimum alongside its actual value, providing quick minimum retrieval by just examining the top of the stack.
Time Complexity: O(1) for each operation.
Space Complexity: O(n), each element only holds a pair so space is efficient relative to the number of elements.
1class MinStack:
2 def __init__(self):
3 self.stack = []
4
5 def push(self, val: int) -> None:
6 if not self.stack:
7 self.stack.append((val, val))
8 else:
9 self.stack.append((val, min(val, self.stack[-1][1])))
10
11 def pop(self) -> None:
12 self.stack.pop()
13
14 def top(self) -> int:
15 return self.stack[-1][0]
16
17 def getMin(self) -> int:
18 return self.stack[-1][1]
19Python takes advantage of tuples to store both the value and current minimum together. When popping or retrieving the top, the value is easily accessible, and minimum retrieval likewise by referring to the second element of the tuple.