343 lines
12 KiB
Java
343 lines
12 KiB
Java
|
|
|
|
/*************************************************************************
|
|
* Compilation: javac IndexMaxPQ.java
|
|
* Execution: java IndexMaxPQ
|
|
*
|
|
* Maximum-oriented indexed PQ implementation using a binary heap.
|
|
*
|
|
*********************************************************************/
|
|
|
|
import java.util.Iterator;
|
|
import java.util.NoSuchElementException;
|
|
|
|
import edu.princeton.cs.introcs.StdOut;
|
|
import edu.princeton.cs.introcs.StdRandom;
|
|
|
|
/**
|
|
* The IndexMaxPQ class represents an indexed priority queue of generic keys.
|
|
* It supports the usual insert and delete-the-maximum
|
|
* operations, along with delete and change-the-key
|
|
* methods. In order to let the client refer to items on the priority queue,
|
|
* an integer between 0 and NMAX-1 is associated with each key—the client
|
|
* uses this integer to specify which key to delete or change.
|
|
* It also supports methods for peeking at a maximum key,
|
|
* testing if the priority queue is empty, and iterating through
|
|
* the keys.
|
|
*
|
|
* This implementation uses a binary heap along with an array to associate
|
|
* keys with integers in the given range.
|
|
* The insert , delete-the-maximum , delete ,
|
|
* change-key , decrease-key , and increase-key
|
|
* operations take logarithmic time.
|
|
* The is-empty , size , max-index , max-key , and key-of
|
|
* operations take constant time.
|
|
* Construction takes time proportional to the specified capacity.
|
|
*
|
|
* For additional documentation, see <a href="http://algs4.cs.princeton.edu/24pq">Section 2.4</a> of
|
|
* Algorithms, 4th Edition by Robert Sedgewick and Kevin Wayne.
|
|
*
|
|
* @author Robert Sedgewick
|
|
* @author Kevin Wayne
|
|
*/
|
|
public class IndexMaxPQ<Key extends Comparable<Key>> implements Iterable<Integer> {
|
|
private int N; // number of elements on PQ
|
|
private int[] pq; // binary heap using 1-based indexing
|
|
private int[] qp; // inverse of pq - qp[pq[i]] = pq[qp[i]] = i
|
|
private Key[] keys; // keys[i] = priority of i
|
|
|
|
/**
|
|
* Initializes an empty indexed priority queue with indices between 0 and NMAX-1.
|
|
* @param NMAX the keys on the priority queue are index from 0 to NMAX-1
|
|
* @throws java.lang.IllegalArgumentException if NMAX < 0
|
|
*/
|
|
public IndexMaxPQ(int NMAX) {
|
|
keys = (Key[]) new Comparable[NMAX + 1]; // make this of length NMAX??
|
|
pq = new int[NMAX + 1];
|
|
qp = new int[NMAX + 1]; // make this of length NMAX??
|
|
for (int i = 0; i <= NMAX; i++) qp[i] = -1;
|
|
}
|
|
|
|
/**
|
|
* Is the priority queue empty?
|
|
* @return true if the priority queue is empty; false otherwise
|
|
*/
|
|
public boolean isEmpty() { return N == 0; }
|
|
|
|
/**
|
|
* Is i an index on the priority queue?
|
|
* @param i an index
|
|
* @throws java.lang.IndexOutOfBoundsException unless (0 ≤ i < NMAX)
|
|
*/
|
|
public boolean contains(int i) {
|
|
return qp[i] != -1;
|
|
}
|
|
|
|
/**
|
|
* Returns the number of keys on the priority queue.
|
|
* @return the number of keys on the priority queue
|
|
*/
|
|
public int size() {
|
|
return N;
|
|
}
|
|
|
|
/**
|
|
* Associate key with index i.
|
|
* @param i an index
|
|
* @param key the key to associate with index i
|
|
* @throws java.lang.IndexOutOfBoundsException unless 0 ≤ i < NMAX
|
|
* @throws java.util.IllegalArgumentException if there already is an item associated with index i
|
|
*/
|
|
public void insert(int i, Key key) {
|
|
if (contains(i)) throw new IllegalArgumentException("index is already in the priority queue");
|
|
N++;
|
|
qp[i] = N;
|
|
pq[N] = i;
|
|
keys[i] = key;
|
|
swim(N);
|
|
}
|
|
|
|
/**
|
|
* Returns an index associated with a maximum key.
|
|
* @return an index associated with a maximum key
|
|
* @throws java.util.NoSuchElementException if priority queue is empty
|
|
*/
|
|
public int maxIndex() {
|
|
if (N == 0) throw new NoSuchElementException("Priority queue underflow");
|
|
return pq[1];
|
|
}
|
|
|
|
/**
|
|
* Return a maximum key.
|
|
* @return a maximum key
|
|
* @throws java.util.NoSuchElementException if priority queue is empty
|
|
*/
|
|
public Key maxKey() {
|
|
if (N == 0) throw new NoSuchElementException("Priority queue underflow");
|
|
return keys[pq[1]];
|
|
}
|
|
|
|
/**
|
|
* Removes a maximum key and returns its associated index.
|
|
* @return an index associated with a maximum key
|
|
* @throws java.util.NoSuchElementException if priority queue is empty
|
|
*/
|
|
public int delMax() {
|
|
if (N == 0) throw new NoSuchElementException("Priority queue underflow");
|
|
int min = pq[1];
|
|
exch(1, N--);
|
|
sink(1);
|
|
qp[min] = -1; // delete
|
|
keys[pq[N+1]] = null; // to help with garbage collection
|
|
pq[N+1] = -1; // not needed
|
|
return min;
|
|
}
|
|
|
|
/**
|
|
* Returns the key associated with index i.
|
|
* @param i the index of the key to return
|
|
* @return the key associated with index i
|
|
* @throws java.lang.IndexOutOfBoundsException unless 0 ≤ i < NMAX
|
|
* @throws java.util.NoSuchElementException no key is associated with index i
|
|
*/
|
|
public Key keyOf(int i) {
|
|
if (!contains(i)) throw new NoSuchElementException("index is not in the priority queue");
|
|
else return keys[i];
|
|
}
|
|
|
|
/**
|
|
* Change the key associated with index i to the specified value.
|
|
* @param i the index of the key to change
|
|
* @param key change the key assocated with index i to this key
|
|
* @throws java.lang.IndexOutOfBoundsException unless 0 ≤ i < NMAX
|
|
* @deprecated Replaced by changeKey()
|
|
*/
|
|
@Deprecated public void change(int i, Key key) {
|
|
changeKey(i, key);
|
|
}
|
|
|
|
/**
|
|
* Change the key associated with index i to the specified value.
|
|
* @param i the index of the key to change
|
|
* @param key change the key assocated with index i to this key
|
|
* @throws java.lang.IndexOutOfBoundsException unless 0 ≤ i < NMAX
|
|
*/
|
|
public void changeKey(int i, Key key) {
|
|
if (!contains(i)) throw new NoSuchElementException("index is not in the priority queue");
|
|
keys[i] = key;
|
|
swim(qp[i]);
|
|
sink(qp[i]);
|
|
}
|
|
|
|
|
|
/**
|
|
* Increase the key associated with index i to the specified value.
|
|
* @param i the index of the key to increase
|
|
* @param key increase the key assocated with index i to this key
|
|
* @throws java.lang.IndexOutOfBoundsException unless 0 ≤ i < NMAX
|
|
* @throws java.lang.IllegalArgumentException if key ≤ key associated with index i
|
|
* @throws java.util.NoSuchElementException no key is associated with index i
|
|
*/
|
|
public void increaseKey(int i, Key key) {
|
|
if (!contains(i)) throw new NoSuchElementException("index is not in the priority queue");
|
|
if (keys[i].compareTo(key) >= 0) throw new IllegalArgumentException("Calling increaseKey() with given argument would not strictly increase the key");
|
|
|
|
|
|
keys[i] = key;
|
|
swim(qp[i]);
|
|
}
|
|
|
|
/**
|
|
* Decrease the key associated with index i to the specified value.
|
|
* @param i the index of the key to decrease
|
|
* @param key decrease the key assocated with index i to this key
|
|
* @throws java.lang.IndexOutOfBoundsException unless 0 ≤ i < NMAX
|
|
* @throws java.lang.IllegalArgumentException if key ≥ key associated with index i
|
|
* @throws java.util.NoSuchElementException no key is associated with index i
|
|
*/
|
|
public void decreaseKey(int i, Key key) {
|
|
if (!contains(i)) throw new NoSuchElementException("index is not in the priority queue");
|
|
if (keys[i].compareTo(key) <= 0) throw new IllegalArgumentException("Calling decreaseKey() with given argument would not strictly decrease the key");
|
|
|
|
keys[i] = key;
|
|
sink(qp[i]);
|
|
}
|
|
|
|
/**
|
|
* Remove the key associated with index i.
|
|
* @param i the index of the key to remove
|
|
* @throws java.lang.IndexOutOfBoundsException unless 0 ≤ i < NMAX
|
|
* @throws java.util.NoSuchElementException no key is associated with index i
|
|
*/
|
|
public void delete(int i) {
|
|
if (!contains(i)) throw new NoSuchElementException("index is not in the priority queue");
|
|
int index = qp[i];
|
|
exch(index, N--);
|
|
swim(index);
|
|
sink(index);
|
|
keys[i] = null;
|
|
qp[i] = -1;
|
|
}
|
|
|
|
|
|
/**************************************************************
|
|
* General helper functions
|
|
**************************************************************/
|
|
private boolean less(int i, int j) {
|
|
return keys[pq[i]].compareTo(keys[pq[j]]) < 0;
|
|
}
|
|
|
|
private void exch(int i, int j) {
|
|
int swap = pq[i]; pq[i] = pq[j]; pq[j] = swap;
|
|
qp[pq[i]] = i; qp[pq[j]] = j;
|
|
}
|
|
|
|
|
|
/**************************************************************
|
|
* Heap helper functions
|
|
**************************************************************/
|
|
private void swim(int k) {
|
|
while (k > 1 && less(k/2, k)) {
|
|
exch(k, k/2);
|
|
k = k/2;
|
|
}
|
|
}
|
|
|
|
private void sink(int k) {
|
|
while (2*k <= N) {
|
|
int j = 2*k;
|
|
if (j < N && less(j, j+1)) j++;
|
|
if (!less(k, j)) break;
|
|
exch(k, j);
|
|
k = j;
|
|
}
|
|
}
|
|
|
|
|
|
/***********************************************************************
|
|
* Iterators
|
|
**********************************************************************/
|
|
|
|
/**
|
|
* Returns an iterator that iterates over the keys on the
|
|
* priority queue in descending order.
|
|
* The iterator doesn't implement remove() since it's optional.
|
|
* @return an iterator that iterates over the keys in descending order
|
|
*/
|
|
public Iterator<Integer> iterator() { return new HeapIterator(); }
|
|
|
|
private class HeapIterator implements Iterator<Integer> {
|
|
// create a new pq
|
|
private IndexMaxPQ<Key> copy;
|
|
|
|
// add all elements to copy of heap
|
|
// takes linear time since already in heap order so no keys move
|
|
public HeapIterator() {
|
|
copy = new IndexMaxPQ<Key>(pq.length - 1);
|
|
for (int i = 1; i <= N; i++)
|
|
copy.insert(pq[i], keys[pq[i]]);
|
|
}
|
|
|
|
public boolean hasNext() { return !copy.isEmpty(); }
|
|
public void remove() { throw new UnsupportedOperationException(); }
|
|
|
|
public Integer next() {
|
|
if (!hasNext()) throw new NoSuchElementException();
|
|
return copy.delMax();
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Unit tests the IndexMaxPQ data type.
|
|
*/
|
|
public static void main(String[] args) {
|
|
// insert a bunch of strings
|
|
String[] strings = { "it", "was", "the", "best", "of", "times", "it", "was", "the", "worst" };
|
|
|
|
IndexMaxPQ<String> pq = new IndexMaxPQ<String>(strings.length);
|
|
for (int i = 0; i < strings.length; i++) {
|
|
pq.insert(i, strings[i]);
|
|
}
|
|
|
|
// print each key using the iterator
|
|
for (int i : pq) {
|
|
StdOut.println(i + " " + strings[i]);
|
|
}
|
|
|
|
StdOut.println();
|
|
|
|
// increase or decrease the key
|
|
for (int i = 0; i < strings.length; i++) {
|
|
if (StdRandom.uniform() < 0.5)
|
|
pq.increaseKey(i, strings[i] + strings[i]);
|
|
else
|
|
pq.decreaseKey(i, strings[i].substring(0, 1));
|
|
}
|
|
|
|
// delete and print each key
|
|
while (!pq.isEmpty()) {
|
|
String key = pq.maxKey();
|
|
int i = pq.delMax();
|
|
StdOut.println(i + " " + key);
|
|
}
|
|
StdOut.println();
|
|
|
|
// reinsert the same strings
|
|
for (int i = 0; i < strings.length; i++) {
|
|
pq.insert(i, strings[i]);
|
|
}
|
|
|
|
// delete them in random order
|
|
int[] perm = new int[strings.length];
|
|
for (int i = 0; i < strings.length; i++)
|
|
perm[i] = i;
|
|
StdRandom.shuffle(perm);
|
|
for (int i = 0; i < perm.length; i++) {
|
|
String key = pq.keyOf(perm[i]);
|
|
pq.delete(perm[i]);
|
|
StdOut.println(perm[i] + " " + key);
|
|
}
|
|
|
|
}
|
|
}
|