实现桟的push和pop操作,以及桟的min操作返回桟中的最小值,要求这三个操作的时间复杂度均为O(1)。
在Java中可以使用LinkedList实现桟的各种操作,这里使用双向链表实现桟的push和pop操作,这两个操作都能维持O(1)的时间复杂度,但是对于求桟中元素的最小值,最容易想到的方法是遍历整个桟,然后返回,但此时的时间复杂度为O(n),要想省下时间复杂度,则必须牺牲空间复杂度,所以可以再维护一个和桟大小相同的数据结构(可以是链表,动态数组或者一个新的桟)来存储每个元素入桟之后对应的桟中的最小元素,在对桟执行push和pop操作时,这个数据结构也跟着变化,这样就能以O(1)的时间复杂度实现min操作了。
public class Stack1<Key extends Comparable<Key>> { private Node top; private Node minTop; private class Node { Key key; Node next; Node prev; Node(Key key, Node next, Node prev) { this.key = key; this.next = next; this.prev = prev; } } /* 元素入桟,因为要求桟的push,pop以及min操作都为O(1)操作,所以 * 在元素入桟时还要维护一个min桟 */ public void push(Key key) { if (top == null) { top = new Node(key, null, null); top.next = null; top.prev = null; pushMin(key); return; } Node x = new Node(key, null, top); top.next = x; top = x; pushMin(key); } /* 维护一个最小元素的桟 */ private void pushMin(Key key) { if (minTop == null) { minTop = new Node(key, null, null); minTop.next = null; return; } if (key.compareTo(minTop.key) < 0) { Node x = new Node(key, null, minTop); minTop.next = x; minTop = x; } else { Node x = new Node(minTop.key, null, minTop); minTop.next = x; minTop = x; } } /* 元素出桟,因为要求桟的push,pop以及min操作都为O(1)操作,所以 * 在元素入桟时还要维护一个min桟 */ public Key pop() { if (top == null) { return null; } Key key = top.key; if ((top.prev == null) && (top.next == null)) { top = null; popMin(); return key; } top = top.prev; top.next = null; popMin(); return key; } /* 返回桟中的最小元素 */ private Key popMin() { if (minTop == null) { return null; } Key key = minTop.key; if ((minTop.prev == null) && (minTop.next == null)) { minTop = null; return key; } minTop = minTop.prev; minTop.next = null; return minTop.key; } /* 返回桟的最小元素 */ public Key min() { /* 注意边界条件,当维护的桟为空时minTop也为空 */ if (minTop == null) { return null; } return minTop.key; } public static void main(String[] args) { Stack1<Integer> stack1 = new Stack1<Integer>(); stack1.push(11); stack1.push(23); stack1.push(4); stack1.push(76); stack1.push(13); stack1.push(42); System.out.println(stack1.min() + " " + stack1.pop()); System.out.println(stack1.min() + " " + stack1.pop()); System.out.println(stack1.min() + " " + stack1.pop()); System.out.println(stack1.min() + " " + stack1.pop()); System.out.println(stack1.min() + " " + stack1.pop()); System.out.println(stack1.min() + " " + stack1.pop()); System.out.println(stack1.min() + " " + stack1.pop()); } }
上面main函数的输出为:
4 42 4 13 4 76 4 4 11 23 11 11 null null