May 24, 2021

Min Stack

Design a stack that supports push, pop, top, and retrieving the minimum element in constant time.

Implement the MinStack class:

Example 1:

Input
["MinStack","push","push","push","getMin","pop","top","getMin"]
[[],[-2],[0],[-3],[],[],[],[]]

Output
[null,null,null,null,-3,null,0,-2]

Explanation
MinStack minStack = new MinStack();
minStack.push(-2);
minStack.push(0);
minStack.push(-3);
minStack.getMin(); // return -3
minStack.pop();
minStack.top();    // return 0
minStack.getMin(); // return -2

Solution:

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
class MinStack {
private List<Integer> values = null;
private List<Integer> mins = null;

public MinStack() {
this.values = new ArrayList<Integer>();
this.mins = new ArrayList<Integer>();
}

private void setMin(int val) {
if(this.mins.isEmpty() || val <= this.getMin()) {
this.mins.add(val);
}
}

public void push(int val) {
this.values.add(val);
setMin(val);
}

public void pop() {
if(this.values.remove(this.values.size() - 1) == this.getMin()) {
this.mins.remove(this.mins.size() - 1);
}
}

public int top() {
return this.values.get(this.values.size() - 1);
}

public int getMin() {
return this.mins.get(this.mins.size() - 1);
}
}

About this Post

This post is written by Mahfuzur Rahman, licensed under CC BY-NC 4.0.

#algorithm#java#stack