《Algorithms 4th Edition》读书笔记——2.4 优先队列(priority queue)-Ⅴ

命题Q。对于一个含有N个元素的基于堆叠优先队列,插入元素操作只需要不超过(lgN + 1)次比较,删除最大元素的操作需要不超过2lgN次比较。

证明。由命题P可知,两种操作都需要在根节点和堆底之间移动元素,而路径的长度不超过lgN。对于路径上的每个节点,删除最大元素需要两次比比较(除了堆底元素),一次用来找出较大的子节点,一次用来确定该子节点是否需要上浮。

 

对于需要大量混杂的插入和删除最大元素操作的典型应用来说,命题Q意味着一个重要的性能突破(详见优先队列增长数量级表)。使用有序或是无序数组的优先队列的初级实现总是需要线性时间来完成其中一种操作,但基于堆底实现则能够抱枕在对数时间内完成他们。这种差别使得我们能够解决以前无法解决的问题。

 

2.4.4.3 多叉堆

 

基于用数组表示的完全三叉树构造堆并修改相应的代码并不困难。对于数组中1 至N 的N个元素,位置k的结点大于等于位于3k - 1、3k和3k + 1的结点,小于等于位于[(k + 1) / 3 (d)]的结点。甚至对于给定的d,将其修改为任意的d叉树也并不困难。我们需要在树高(logaN)和在每个节点的d个子节点找到最大者的代价之间找到折中,这取决于实现的细节以及不同操作的预期相对频繁程度。堆上的优先队列操作如右图。

2.4.4.4 调整数组大小

 

我们可以添加一个没有参数的构造函数,在insert()中添加将数组长度加倍的代码,在delMax()中添加将数组长度减半的代码,就像栈那样。这样,算法的用例就无需管组各种队列大小的限制。当有限队列的数组大小可以跳帧、队列长度可以是任意值时,命题Q指出的对数时间复杂度上限就只是针对一般性的队列长度N而言了。

 

2.4.4.5 元素的不可变性

 

有限队列存储了用例创建的对象,但同时假设用例代码不会改变他们(改变他们就可能打破堆的有序性)。我们可以将这个假设为强制条件,但程序员通常不会这么做,因为增加代码的复杂性会降低性能。

 

2.4.4.6 索引优先队列

 

在很多应用中,允许用例引用已经进入有限队列中的元素是有必要的。做到这一点的一种简单方法是用例已经有了总量为N的多个元素,而且可能还同时使用了多个(平行)数组(Parallel Array)来存储这些元素的信息。此时,其他无关的用例代码可能已经在使用一个整数索引来引用这些元素了。这些考虑引导我们设计了下表。

理解这种数据结构的一个较好方法是将它看成一个能够快速访问其中最小元素的数组。事实上它还要更好——它能够快速访问数组的一个特定子集中的最小元素(指所有被插入的元素)。换句话说,可以将名为pq的IndexMinPQ优先队列看做数组pq[0..N - 1]中的一部分元素的代表。将pq.insert(k, item)看做将k加入这个子集并使pq[k] = item, pq.change(k, item)则代表令pq[k] = item。这两种操作没有改变其他操作所依赖的数据结构,其中最重要的就是delMin()(删除最小元素并返回它的索引)和change()(改变数据结构中的某个元素的索引——即pq[i] = item)。这些操作在许多应用中都很重要并且依赖于对元素的引用(索引)。一般来说,当堆发生变化时,我们会用下沉(元素减小时)或上浮(元素变大时)操作来恢复堆的有序性。在这些操作中,我们可以用索引查找元素。能够定位堆中的任意元素也使我们能够在API中加入一个delete()操作。

 

命题Q(续)。在一个大小为N的索引优先队列中,插入元素(insert)、改变优先级(change)、删除(delete)和删除最大小元素(remove the minimum)操作所需的比较次数和logN成正比(如后表)

证明。已知堆中所有路径最长即为~lgN,从代码中很容易得到这个结论。

 

操作 比较次数的增长数量级
insert() logN
change() logN
contains() 1
delete() logN
min() 1
minIndex() 1
delMin

logN

这段讨论针对的是找好粗最小元素的队列;以下是《alg4》书中实现的一个找出最大元素的版本IndexMaxPQ。

  1 import java.util.Iterator;
  2 import java.util.NoSuchElementException;
  3 
  4 /**
  5  *  The <tt>IndexMaxPQ</tt> class represents an indexed priority queue of generic keys.
  6  *  It supports the usual <em>insert</em> and <em>delete-the-maximum</em>
  7  *  operations, along with <em>delete</em> and <em>change-the-key</em> 
  8  *  methods. In order to let the client refer to items on the priority queue,
  9  *  an integer between 0 and NMAX-1 is associated with each key&mdash;the client
 10  *  uses this integer to specify which key to delete or change.
 11  *  It also supports methods for peeking at a maximum key,
 12  *  testing if the priority queue is empty, and iterating through
 13  *  the keys.
 14  *  <p>
 15  *  This implementation uses a binary heap along with an array to associate
 16  *  keys with integers in the given range.
 17  *  The <em>insert</em>, <em>delete-the-maximum</em>, <em>delete</em>,
 18  *  <em>change-key</em>, <em>decrease-key</em>, and <em>increase-key</em>
 19  *  operations take logarithmic time.
 20  *  The <em>is-empty</em>, <em>size</em>, <em>max-index</em>, <em>max-key</em>, and <em>key-of</em>
 21  *  operations take constant time.
 22  *  Construction takes time proportional to the specified capacity.
 23  *  <p>
 24  *  For additional documentation, see <a href="http://algs4.cs.princeton.edu/24pq">Section 2.4</a> of
 25  *  <i>Algorithms, 4th Edition</i> by Robert Sedgewick and Kevin Wayne.
 26  *
 27  *  @author Robert Sedgewick
 28  *  @author Kevin Wayne
 29  */
 30 public class IndexMaxPQ<Key extends Comparable<Key>> implements Iterable<Integer> {
 31     private int N;           // number of elements on PQ
 32     private int[] pq;        // binary heap using 1-based indexing
 33     private int[] qp;        // inverse of pq - qp[pq[i]] = pq[qp[i]] = i
 34     private Key[] keys;      // keys[i] = priority of i
 35 
 36     /**
 37      * Initializes an empty indexed priority queue with indices between 0 and NMAX-1.
 38      * @param NMAX the keys on the priority queue are index from 0 to NMAX-1
 39      * @throws java.lang.IllegalArgumentException if NMAX < 0
 40      */
 41     public IndexMaxPQ(int NMAX) {
 42         keys = (Key[]) new Comparable[NMAX + 1];    // make this of length NMAX??
 43         pq   = new int[NMAX + 1];
 44         qp   = new int[NMAX + 1];                   // make this of length NMAX??
 45         for (int i = 0; i <= NMAX; i++) qp[i] = -1;
 46     }
 47 
 48     /**
 49      * Is the priority queue empty?
 50      * @return true if the priority queue is empty; false otherwise
 51      */
 52     public boolean isEmpty() { return N == 0; }
 53 
 54     /**
 55      * Is i an index on the priority queue?
 56      * @param i an index
 57      * @throws java.lang.IndexOutOfBoundsException unless (0 &le; i < NMAX)
 58      */
 59     public boolean contains(int i) {
 60         return qp[i] != -1;
 61     }
 62 
 63     /**
 64      * Returns the number of keys on the priority queue.
 65      * @return the number of keys on the priority queue 
 66      */
 67     public int size() {
 68         return N;
 69     }
 70 
 71    /**
 72      * Associate key with index i.
 73      * @param i an index
 74      * @param key the key to associate with index i
 75      * @throws java.lang.IndexOutOfBoundsException unless 0 &le; i < NMAX
 76      * @throws java.util.IllegalArgumentException if there already is an item associated with index i
 77      */
 78     public void insert(int i, Key key) {
 79         if (contains(i)) throw new IllegalArgumentException("index is already in the priority queue");
 80         N++;
 81         qp[i] = N;
 82         pq[N] = i;
 83         keys[i] = key;
 84         swim(N);
 85     }
 86 
 87     /**
 88      * Returns an index associated with a maximum key.
 89      * @return an index associated with a maximum key
 90      * @throws java.util.NoSuchElementException if priority queue is empty
 91      */
 92     public int maxIndex() { 
 93         if (N == 0) throw new NoSuchElementException("Priority queue underflow");
 94         return pq[1];
 95     }
 96 
 97     /**
 98      * Return a maximum key.
 99      * @return a maximum key
100      * @throws java.util.NoSuchElementException if priority queue is empty
101      */
102     public Key maxKey() { 
103         if (N == 0) throw new NoSuchElementException("Priority queue underflow");
104         return keys[pq[1]];
105     }
106 
107     /**
108      * Removes a maximum key and returns its associated index.
109      * @return an index associated with a maximum key
110      * @throws java.util.NoSuchElementException if priority queue is empty
111      */
112     public int delMax() { 
113         if (N == 0) throw new NoSuchElementException("Priority queue underflow");
114         int min = pq[1];        
115         exch(1, N--); 
116         sink(1);
117         qp[min] = -1;            // delete
118         keys[pq[N+1]] = null;    // to help with garbage collection
119         pq[N+1] = -1;            // not needed
120         return min; 
121     }
122 
123     /**
124      * Returns the key associated with index i.
125      * @param i the index of the key to return
126      * @return the key associated with index i
127      * @throws java.lang.IndexOutOfBoundsException unless 0 &le; i < NMAX
128      * @throws java.util.NoSuchElementException no key is associated with index i
129      */
130     public Key keyOf(int i) {
131         if (!contains(i)) throw new NoSuchElementException("index is not in the priority queue");
132         else return keys[i];
133     }
134 
135    /**
136      * Change the key associated with index i to the specified value.
137      * @param i the index of the key to change
138      * @param key change the key assocated with index i to this key
139      * @throws java.lang.IndexOutOfBoundsException unless 0 &le; i < NMAX
140      * @deprecated Replaced by changeKey()
141      */
142     @Deprecated public void change(int i, Key key) {
143         changeKey(i, key);
144     }
145 
146     /**
147      * Change the key associated with index i to the specified value.
148      * @param i the index of the key to change
149      * @param key change the key assocated with index i to this key
150      * @throws java.lang.IndexOutOfBoundsException unless 0 &le; i < NMAX
151      */
152     public void changeKey(int i, Key key) {
153         if (!contains(i)) throw new NoSuchElementException("index is not in the priority queue");
154         keys[i] = key;
155         swim(qp[i]);
156         sink(qp[i]);
157     }
158 
159 
160     /**
161      * Increase the key associated with index i to the specified value.
162      * @param i the index of the key to increase
163      * @param key increase the key assocated with index i to this key
164      * @throws java.lang.IndexOutOfBoundsException unless 0 &le; i < NMAX
165      * @throws java.lang.IllegalArgumentException if key &le; key associated with index i
166      * @throws java.util.NoSuchElementException no key is associated with index i
167      */
168     public void increaseKey(int i, Key key) {
169         if (!contains(i)) throw new NoSuchElementException("index is not in the priority queue");
170         if (keys[i].compareTo(key) >= 0) throw new IllegalArgumentException("Calling increaseKey() with given argument would not strictly increase the key");
171 
172 
173         keys[i] = key;
174         swim(qp[i]);
175     }
176 
177     /**
178      * Decrease the key associated with index i to the specified value.
179      * @param i the index of the key to decrease
180      * @param key decrease the key assocated with index i to this key
181      * @throws java.lang.IndexOutOfBoundsException unless 0 &le; i < NMAX
182      * @throws java.lang.IllegalArgumentException if key &ge; key associated with index i
183      * @throws java.util.NoSuchElementException no key is associated with index i
184      */
185     public void decreaseKey(int i, Key key) {
186         if (!contains(i)) throw new NoSuchElementException("index is not in the priority queue");
187         if (keys[i].compareTo(key) <= 0) throw new IllegalArgumentException("Calling decreaseKey() with given argument would not strictly decrease the key");
188 
189         keys[i] = key;
190         sink(qp[i]);
191     }
192 
193     /**
194      * Remove the key associated with index i.
195      * @param i the index of the key to remove
196      * @throws java.lang.IndexOutOfBoundsException unless 0 &le; i < NMAX
197      * @throws java.util.NoSuchElementException no key is associated with index i
198      */
199     public void delete(int i) {
200         if (!contains(i)) throw new NoSuchElementException("index is not in the priority queue");
201         int index = qp[i];
202         exch(index, N--);
203         swim(index);
204         sink(index);
205         keys[i] = null;
206         qp[i] = -1;
207     }
208 
209 
210    /**************************************************************
211     * General helper functions
212     **************************************************************/
213     private boolean less(int i, int j) {
214         return keys[pq[i]].compareTo(keys[pq[j]]) < 0;
215     }
216 
217     private void exch(int i, int j) {
218         int swap = pq[i]; pq[i] = pq[j]; pq[j] = swap;
219         qp[pq[i]] = i; qp[pq[j]] = j;
220     }
221 
222 
223    /**************************************************************
224     * Heap helper functions
225     **************************************************************/
226     private void swim(int k)  {
227         while (k > 1 && less(k/2, k)) {
228             exch(k, k/2);
229             k = k/2;
230         }
231     }
232 
233     private void sink(int k) {
234         while (2*k <= N) {
235             int j = 2*k;
236             if (j < N && less(j, j+1)) j++;
237             if (!less(k, j)) break;
238             exch(k, j);
239             k = j;
240         }
241     }
242 
243 
244    /***********************************************************************
245     * Iterators
246     **********************************************************************/
247 
248     /**
249      * Returns an iterator that iterates over the keys on the
250      * priority queue in descending order.
251      * The iterator doesn't implement <tt>remove()</tt> since it's optional.
252      * @return an iterator that iterates over the keys in descending order
253      */
254     public Iterator<Integer> iterator() { return new HeapIterator(); }
255 
256     private class HeapIterator implements Iterator<Integer> {
257         // create a new pq
258         private IndexMaxPQ<Key> copy;
259 
260         // add all elements to copy of heap
261         // takes linear time since already in heap order so no keys move
262         public HeapIterator() {
263             copy = new IndexMaxPQ<Key>(pq.length - 1);
264             for (int i = 1; i <= N; i++)
265                 copy.insert(pq[i], keys[pq[i]]);
266         }
267 
268         public boolean hasNext()  { return !copy.isEmpty();                     }
269         public void remove()      { throw new UnsupportedOperationException();  }
270 
271         public Integer next() {
272             if (!hasNext()) throw new NoSuchElementException();
273             return copy.delMax();
274         }
275     }
276 
277     /**
278      * Unit tests the <tt>IndexMaxPQ</tt> data type.
279      */
280     public static void main(String[] args) {
281         // insert a bunch of strings
282         String[] strings = { "it", "was", "the", "best", "of", "times", "it", "was", "the", "worst" };
283 
284         IndexMaxPQ<String> pq = new IndexMaxPQ<String>(strings.length);
285         for (int i = 0; i < strings.length; i++) {
286             pq.insert(i, strings[i]);
287         }
288 
289         // print each key using the iterator
290         for (int i : pq) {
291             StdOut.println(i + " " + strings[i]);
292         }
293 
294         StdOut.println();
295 
296         // increase or decrease the key
297         for (int i = 0; i < strings.length; i++) {
298             if (StdRandom.uniform() < 0.5)
299                 pq.increaseKey(i, strings[i] + strings[i]);
300             else
301                 pq.decreaseKey(i, strings[i].substring(0, 1));
302         }
303 
304         // delete and print each key
305         while (!pq.isEmpty()) {
306             String key = pq.maxKey();
307             int i = pq.delMax();
308             StdOut.println(i + " " + key);
309         }
310         StdOut.println();
311 
312         // reinsert the same strings
313         for (int i = 0; i < strings.length; i++) {
314             pq.insert(i, strings[i]);
315         }
316 
317         // delete them in random order
318         int[] perm = new int[strings.length];
319         for (int i = 0; i < strings.length; i++)
320             perm[i] = i;
321         StdRandom.shuffle(perm);
322         for (int i = 0; i < perm.length; i++) {
323             String key = pq.keyOf(perm[i]);
324             pq.delete(perm[i]);
325             StdOut.println(perm[i] + " " + key);
326         }
327 
328     }
329 }
IndexMaxPQ.java

2.4.4.7 索引有限队列用例

 

下面的用例调用了IndexMinPQ的代码Multiway解决了多项归并问题:它将多个有序的输入流归并成一个有序的输出流。许多应用中都会遇到这个问题。输入可能你来自多种科学仪器的输出(按照时间排序),或是来自多个音乐或电影网站的信息列表(按名称或艺术家名字排列),或是商业交易(按账号或时间排序),或者其他。如果有足够的空间,你可以把它们简单地读入一个数组并排序,但如果用例优先队列,无论输入有多长你都可以把它们全部读入并排序。

 1 /**
 2  *  The <tt>Multiway</tt> class provides a client for reading in several
 3  *  sorted text files and merging them together into a single sorted
 4  *  text stream.
 5  *  This implementation uses a {@link IndexMinPQ} to perform the multiway
 6  *  merge. 
 7  *  <p>
 8  *  For additional documentation, see <a href="http://algs4.cs.princeton.edu/24pq">Section 2.4</a>
 9  *  of <i>Algorithms, 4th Edition</i> by Robert Sedgewick and Kevin Wayne.
10  *
11  *  @author Robert Sedgewick
12  *  @author Kevin Wayne
13  */
14 
15 public class Multiway { 
16 
17     // This class should not be instantiated.
18     private Multiway() { }
19 
20     // merge together the sorted input streams and write the sorted result to standard output
21     private static void merge(In[] streams) { 
22         int N = streams.length; 
23         IndexMinPQ<String> pq = new IndexMinPQ<String>(N); 
24         for (int i = 0; i < N; i++) 
25             if (!streams[i].isEmpty()) 
26                 pq.insert(i, streams[i].readString()); 
27 
28         // Extract and print min and read next from its stream. 
29         while (!pq.isEmpty()) {
30             StdOut.print(pq.minKey() + " "); 
31             int i = pq.delMin(); 
32             if (!streams[i].isEmpty()) 
33                 pq.insert(i, streams[i].readString()); 
34         }
35         StdOut.println();
36     } 
37 
38 
39     /**
40      *  Reads sorted text files specified as command-line arguments;
41      *  merges them together into a sorted output; and writes
42      *  the results to standard output.
43      *  Note: this client does not check that the input files are sorted.
44      */
45     public static void main(String[] args) { 
46         int N = args.length; 
47         In[] streams = new In[N]; 
48         for (int i = 0; i < N; i++) 
49             streams[i] = new In(args[i]); 
50         merge(streams); 
51     } 
52 } 
Multiway.java

 

这段代码调用了IndexMinPQ()来将作为命令行参数输入的多行有序字符串归并为一行有序的输出。每个输入流的索引都关联着一个元素(输入中的下个字符串)。初始化之后,代码进入一个循环,删除并打印出队列中最小的字符串,然后将该输入的下一个字符串添加为一个元素。


原文地址:https://www.cnblogs.com/Destiny-Gem/p/3800069.html