题目:设计包含min 函数的栈。
- 定义栈的数据结构,要求添加一个min 函数,能够得到栈的最小元素。
- 要求函数min、push 以及pop 的时间复杂度都是O(1)。
思路:定义两个栈,一个用来储存数据,一个用来储存最小值。
测试用例:
- 新压入栈中的数字比之前的最小值大
- 新压入栈中的数字比之前的最小值小
- 弹出的数字不是最小元素
- 弹出的数字是最小元素
public class test_thirty {
Stack<Integer> stack_data = new Stack<>();
Stack<Integer> stack_min = new Stack<>();
public void push(int node) {
stack_data.push(node);
if (stack_min.isEmpty()){
stack_min.push(node);
}
else if (node < min()){
stack_min.push(node);
}
else stack_min.push(min());
}
public void pop() {
stack_data.pop();
stack_min.pop();
}
public int top() {
return stack_data.peek();
}
public int min() {
return stack_min.peek();
}
}