[剑指offer] 包含min函数的栈

题目描述

定义栈的数据结构,请在该类型中实现一个能够得到栈最小元素的min函数。

解题思路

用一个栈stack保存数据,用另外一个栈temp保存依次入栈最小的数

比如,stack中依次入栈

5, 3, 4, 10, 2, 12, 1, 8

则temp依次入栈

5, 3, 3,3, 2, 2, 1, 1

每次入栈的时候,如果入栈的元素比min中的栈顶元素小或等于则入栈,否则用最小元素入栈。

参考代码

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
import java.util.Stack;

public class Solution {
Stack<Integer> stack = new Stack<>();
Stack<Integer> temp = new Stack<>();
int min = Integer.MAX_VALUE;
public void push(int node) {
stack.push(node);
if(node < min){
temp.push(node);
min = node;
}
else
temp.push(min);
}

public void pop() {
stack.pop();
temp.pop();
}

public int top() {
int t = stack.pop();
stack.push(t);
return t;
}

public int min() {
int m = temp.pop();
temp.push(m);
return m;
}
}