题目:
定义栈的数据结构,请在该类型中实现一个能够得到栈中所含最小元素的min函数(时间复杂度应为O(1))。
思路:
用一个栈 data 保存数据,用另外一个栈 min 保存依次入栈最小的数
比如,data中依次入栈,5, 4, 3, 8, 10, 11, 12, 1
则 min 依次入栈,5, 4, 3,no,no, no, no, 1
no 代表此次不如栈 每次入栈的时候,如果入栈的元素比 min 中的栈顶元素小或等于则入栈,否则不如栈。
弹栈时,数据栈依次弹出,当最小值栈栈顶元素与数据栈栈顶元素相同时,最小值栈弹栈。
代码:
package 包含min函数的栈;
import java.util.Stack;
/**
* 定义栈的数据结构,请在该类型中实现一个能够得到栈中所含最小元素的min函数(时间复杂度应为O(1))。
* 思路:用一个栈 data 保存数据,用另外一个栈 min 保存依次入栈最小的数
* 比如,data中依次入栈,5, 4, 3, 8, 10, 11, 12, 1
* 则 min 依次入栈,5, 4, 3,no,no, no, no, 1
*
* no 代表此次不如栈 每次入栈的时候,如果入栈的元素比 min 中的栈顶元素小或等于则入栈,否则不如栈。
*
* @author Administrator
*
*/
public class Demo {
// 数据栈,用于存放插入的数据
Stack<Integer> data = new Stack<>();
// 最小数栈,存放数据栈中最小的数的位置
Stack<Integer> min = new Stack<>();
Integer temp = null;
public void push(int node) {
// 数据栈存放插入数据
data.push(node);
// 最小数栈为空时存放第一个插入的数据
if (min.isEmpty()) {
min.push(node);
} else {
// 插入数据小于最小数栈的栈顶元素时,插入最小数栈
if (min.peek() > node)
min.push(node);
}
}
public void pop() {
// 数据栈与最小值栈的栈顶元素相等时,此栈顶元素为最小值
if (data.pop() == min.peek()) {
min.pop();
}
}
public int top() {
// 返回数据栈的栈顶元素
return data.peek();
}
public int min() {
// 返回最小值栈的栈顶元素,即最小值
return min.peek();
}
public static void main(String[] args) {
Demo demo = new Demo();
demo.push(3);
System.out.println(demo.min() == 3);
demo.push(4);
System.out.println(demo.min() == 3);
demo.push(2);
System.out.println(demo.min() == 2);
demo.push(3);
System.out.println(demo.min() == 2);
demo.pop();
System.out.println(demo.min() == 2);
demo.pop();
System.out.println(demo.min() == 3);
demo.push(0);
System.out.println(demo.min() == 0);
}
}