leetcode-包含min函数的栈
题目链接:添加链接描述
题目大意就是定义一个栈,这个栈除了满足基本的栈的效果外,还应该能够返回当前所有栈中元素的最小值。最直接的做法是定义两个栈,一个栈正常存储元素,另外一个栈存储最小的元素。对于第二个栈而言,保证当压入第一个栈的元素小于第二个栈栈顶元素时,才让新元素压入第二个栈,当第一个栈元素出栈时,对比一下,如果出栈的元素是第二个栈的栈顶元素,那么此时整个栈的最小值发生变化。
Stack<Integer> A, B;
public MinStack() {
A = new Stack<>();
B = new Stack<>();
}
public void push(int x) {
A.add(x);
if(B.empty() || B.peek() >= x)
B.add(x);
}
public void pop() {
if(A.pop().equals(B.peek()))
B.pop();
}
public int top() {
return A.peek();
}
public int min() {
return B.peek();
}
如此自然是可以的,不过在评论区看到了一个更有趣的解法,首先自己定义一个链表,在链表的节点上加一个属性min,标记一下当前的最小值。
class ListNode {
int val;
int min;
ListNode next;
ListNode(int x, int minVal, ListNode nextNode) {
val = x;
min = minVal;
next = nextNode;
}
}
对于每个元素,采用头插法进入链表,而且通过对比前一个加入链表的节点的Min来更新自身的min,然后再元素出栈时移除头节点所代表的元素,而整个队列的最小值就是每次的头节点的
min属性。
class MinStack {
ListNode head = null;
/** initialize your data structure here. */
public MinStack() {
}
public void push(int x) {
if(head == null) {
head = new ListNode(x, x, null);
}else {
//把每一个元素插到头部位置,每一个节点记录自身之前的所有元素的最小值
head = new ListNode(x, x<head.min ? x : head.min, head);
}
}
public void pop() {
head = head.next;
}
public int top() {
return head.val;
}
public int min() {
return head.min;
}
}
class ListNode {
int val;
int min;
ListNode next;
ListNode(int x, int minVal, ListNode nextNode) {
val = x;
min = minVal;
next = nextNode;
}
}