jdk/src/share/classes/java/util/PriorityQueue.java
author martin
Sun, 09 May 2010 00:59:49 -0700
changeset 5467 0cfa1c70b5ab
parent 5466 f130bb07764b
child 5506 202f599c92aa
permissions -rw-r--r--
6950540: PriorityQueue(collection) should throw NPE if collection contains a null Summary: Rewrite PriorityQueue constructors for best performance and error handling Reviewed-by: dholmes, chegar
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
2
90ce3da70b43 Initial load
duke
parents:
diff changeset
     1
/*
90ce3da70b43 Initial load
duke
parents:
diff changeset
     2
 * Copyright 2003-2006 Sun Microsystems, Inc.  All Rights Reserved.
90ce3da70b43 Initial load
duke
parents:
diff changeset
     3
 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
90ce3da70b43 Initial load
duke
parents:
diff changeset
     4
 *
90ce3da70b43 Initial load
duke
parents:
diff changeset
     5
 * This code is free software; you can redistribute it and/or modify it
90ce3da70b43 Initial load
duke
parents:
diff changeset
     6
 * under the terms of the GNU General Public License version 2 only, as
90ce3da70b43 Initial load
duke
parents:
diff changeset
     7
 * published by the Free Software Foundation.  Sun designates this
90ce3da70b43 Initial load
duke
parents:
diff changeset
     8
 * particular file as subject to the "Classpath" exception as provided
90ce3da70b43 Initial load
duke
parents:
diff changeset
     9
 * by Sun in the LICENSE file that accompanied this code.
90ce3da70b43 Initial load
duke
parents:
diff changeset
    10
 *
90ce3da70b43 Initial load
duke
parents:
diff changeset
    11
 * This code is distributed in the hope that it will be useful, but WITHOUT
90ce3da70b43 Initial load
duke
parents:
diff changeset
    12
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
90ce3da70b43 Initial load
duke
parents:
diff changeset
    13
 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
90ce3da70b43 Initial load
duke
parents:
diff changeset
    14
 * version 2 for more details (a copy is included in the LICENSE file that
90ce3da70b43 Initial load
duke
parents:
diff changeset
    15
 * accompanied this code).
90ce3da70b43 Initial load
duke
parents:
diff changeset
    16
 *
90ce3da70b43 Initial load
duke
parents:
diff changeset
    17
 * You should have received a copy of the GNU General Public License version
90ce3da70b43 Initial load
duke
parents:
diff changeset
    18
 * 2 along with this work; if not, write to the Free Software Foundation,
90ce3da70b43 Initial load
duke
parents:
diff changeset
    19
 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
90ce3da70b43 Initial load
duke
parents:
diff changeset
    20
 *
90ce3da70b43 Initial load
duke
parents:
diff changeset
    21
 * Please contact Sun Microsystems, Inc., 4150 Network Circle, Santa Clara,
90ce3da70b43 Initial load
duke
parents:
diff changeset
    22
 * CA 95054 USA or visit www.sun.com if you need additional information or
90ce3da70b43 Initial load
duke
parents:
diff changeset
    23
 * have any questions.
90ce3da70b43 Initial load
duke
parents:
diff changeset
    24
 */
90ce3da70b43 Initial load
duke
parents:
diff changeset
    25
90ce3da70b43 Initial load
duke
parents:
diff changeset
    26
package java.util;
90ce3da70b43 Initial load
duke
parents:
diff changeset
    27
90ce3da70b43 Initial load
duke
parents:
diff changeset
    28
/**
90ce3da70b43 Initial load
duke
parents:
diff changeset
    29
 * An unbounded priority {@linkplain Queue queue} based on a priority heap.
90ce3da70b43 Initial load
duke
parents:
diff changeset
    30
 * The elements of the priority queue are ordered according to their
90ce3da70b43 Initial load
duke
parents:
diff changeset
    31
 * {@linkplain Comparable natural ordering}, or by a {@link Comparator}
90ce3da70b43 Initial load
duke
parents:
diff changeset
    32
 * provided at queue construction time, depending on which constructor is
90ce3da70b43 Initial load
duke
parents:
diff changeset
    33
 * used.  A priority queue does not permit {@code null} elements.
90ce3da70b43 Initial load
duke
parents:
diff changeset
    34
 * A priority queue relying on natural ordering also does not permit
90ce3da70b43 Initial load
duke
parents:
diff changeset
    35
 * insertion of non-comparable objects (doing so may result in
90ce3da70b43 Initial load
duke
parents:
diff changeset
    36
 * {@code ClassCastException}).
90ce3da70b43 Initial load
duke
parents:
diff changeset
    37
 *
90ce3da70b43 Initial load
duke
parents:
diff changeset
    38
 * <p>The <em>head</em> of this queue is the <em>least</em> element
90ce3da70b43 Initial load
duke
parents:
diff changeset
    39
 * with respect to the specified ordering.  If multiple elements are
90ce3da70b43 Initial load
duke
parents:
diff changeset
    40
 * tied for least value, the head is one of those elements -- ties are
90ce3da70b43 Initial load
duke
parents:
diff changeset
    41
 * broken arbitrarily.  The queue retrieval operations {@code poll},
90ce3da70b43 Initial load
duke
parents:
diff changeset
    42
 * {@code remove}, {@code peek}, and {@code element} access the
90ce3da70b43 Initial load
duke
parents:
diff changeset
    43
 * element at the head of the queue.
90ce3da70b43 Initial load
duke
parents:
diff changeset
    44
 *
90ce3da70b43 Initial load
duke
parents:
diff changeset
    45
 * <p>A priority queue is unbounded, but has an internal
90ce3da70b43 Initial load
duke
parents:
diff changeset
    46
 * <i>capacity</i> governing the size of an array used to store the
90ce3da70b43 Initial load
duke
parents:
diff changeset
    47
 * elements on the queue.  It is always at least as large as the queue
90ce3da70b43 Initial load
duke
parents:
diff changeset
    48
 * size.  As elements are added to a priority queue, its capacity
90ce3da70b43 Initial load
duke
parents:
diff changeset
    49
 * grows automatically.  The details of the growth policy are not
90ce3da70b43 Initial load
duke
parents:
diff changeset
    50
 * specified.
90ce3da70b43 Initial load
duke
parents:
diff changeset
    51
 *
90ce3da70b43 Initial load
duke
parents:
diff changeset
    52
 * <p>This class and its iterator implement all of the
90ce3da70b43 Initial load
duke
parents:
diff changeset
    53
 * <em>optional</em> methods of the {@link Collection} and {@link
90ce3da70b43 Initial load
duke
parents:
diff changeset
    54
 * Iterator} interfaces.  The Iterator provided in method {@link
90ce3da70b43 Initial load
duke
parents:
diff changeset
    55
 * #iterator()} is <em>not</em> guaranteed to traverse the elements of
90ce3da70b43 Initial load
duke
parents:
diff changeset
    56
 * the priority queue in any particular order. If you need ordered
90ce3da70b43 Initial load
duke
parents:
diff changeset
    57
 * traversal, consider using {@code Arrays.sort(pq.toArray())}.
90ce3da70b43 Initial load
duke
parents:
diff changeset
    58
 *
90ce3da70b43 Initial load
duke
parents:
diff changeset
    59
 * <p> <strong>Note that this implementation is not synchronized.</strong>
90ce3da70b43 Initial load
duke
parents:
diff changeset
    60
 * Multiple threads should not access a {@code PriorityQueue}
90ce3da70b43 Initial load
duke
parents:
diff changeset
    61
 * instance concurrently if any of the threads modifies the queue.
90ce3da70b43 Initial load
duke
parents:
diff changeset
    62
 * Instead, use the thread-safe {@link
90ce3da70b43 Initial load
duke
parents:
diff changeset
    63
 * java.util.concurrent.PriorityBlockingQueue} class.
90ce3da70b43 Initial load
duke
parents:
diff changeset
    64
 *
90ce3da70b43 Initial load
duke
parents:
diff changeset
    65
 * <p>Implementation note: this implementation provides
90ce3da70b43 Initial load
duke
parents:
diff changeset
    66
 * O(log(n)) time for the enqueing and dequeing methods
90ce3da70b43 Initial load
duke
parents:
diff changeset
    67
 * ({@code offer}, {@code poll}, {@code remove()} and {@code add});
90ce3da70b43 Initial load
duke
parents:
diff changeset
    68
 * linear time for the {@code remove(Object)} and {@code contains(Object)}
90ce3da70b43 Initial load
duke
parents:
diff changeset
    69
 * methods; and constant time for the retrieval methods
90ce3da70b43 Initial load
duke
parents:
diff changeset
    70
 * ({@code peek}, {@code element}, and {@code size}).
90ce3da70b43 Initial load
duke
parents:
diff changeset
    71
 *
90ce3da70b43 Initial load
duke
parents:
diff changeset
    72
 * <p>This class is a member of the
90ce3da70b43 Initial load
duke
parents:
diff changeset
    73
 * <a href="{@docRoot}/../technotes/guides/collections/index.html">
90ce3da70b43 Initial load
duke
parents:
diff changeset
    74
 * Java Collections Framework</a>.
90ce3da70b43 Initial load
duke
parents:
diff changeset
    75
 *
90ce3da70b43 Initial load
duke
parents:
diff changeset
    76
 * @since 1.5
90ce3da70b43 Initial load
duke
parents:
diff changeset
    77
 * @author Josh Bloch, Doug Lea
90ce3da70b43 Initial load
duke
parents:
diff changeset
    78
 * @param <E> the type of elements held in this collection
90ce3da70b43 Initial load
duke
parents:
diff changeset
    79
 */
90ce3da70b43 Initial load
duke
parents:
diff changeset
    80
public class PriorityQueue<E> extends AbstractQueue<E>
90ce3da70b43 Initial load
duke
parents:
diff changeset
    81
    implements java.io.Serializable {
90ce3da70b43 Initial load
duke
parents:
diff changeset
    82
90ce3da70b43 Initial load
duke
parents:
diff changeset
    83
    private static final long serialVersionUID = -7720805057305804111L;
90ce3da70b43 Initial load
duke
parents:
diff changeset
    84
90ce3da70b43 Initial load
duke
parents:
diff changeset
    85
    private static final int DEFAULT_INITIAL_CAPACITY = 11;
90ce3da70b43 Initial load
duke
parents:
diff changeset
    86
90ce3da70b43 Initial load
duke
parents:
diff changeset
    87
    /**
90ce3da70b43 Initial load
duke
parents:
diff changeset
    88
     * Priority queue represented as a balanced binary heap: the two
90ce3da70b43 Initial load
duke
parents:
diff changeset
    89
     * children of queue[n] are queue[2*n+1] and queue[2*(n+1)].  The
90ce3da70b43 Initial load
duke
parents:
diff changeset
    90
     * priority queue is ordered by comparator, or by the elements'
90ce3da70b43 Initial load
duke
parents:
diff changeset
    91
     * natural ordering, if comparator is null: For each node n in the
90ce3da70b43 Initial load
duke
parents:
diff changeset
    92
     * heap and each descendant d of n, n <= d.  The element with the
90ce3da70b43 Initial load
duke
parents:
diff changeset
    93
     * lowest value is in queue[0], assuming the queue is nonempty.
90ce3da70b43 Initial load
duke
parents:
diff changeset
    94
     */
90ce3da70b43 Initial load
duke
parents:
diff changeset
    95
    private transient Object[] queue;
90ce3da70b43 Initial load
duke
parents:
diff changeset
    96
90ce3da70b43 Initial load
duke
parents:
diff changeset
    97
    /**
90ce3da70b43 Initial load
duke
parents:
diff changeset
    98
     * The number of elements in the priority queue.
90ce3da70b43 Initial load
duke
parents:
diff changeset
    99
     */
90ce3da70b43 Initial load
duke
parents:
diff changeset
   100
    private int size = 0;
90ce3da70b43 Initial load
duke
parents:
diff changeset
   101
90ce3da70b43 Initial load
duke
parents:
diff changeset
   102
    /**
90ce3da70b43 Initial load
duke
parents:
diff changeset
   103
     * The comparator, or null if priority queue uses elements'
90ce3da70b43 Initial load
duke
parents:
diff changeset
   104
     * natural ordering.
90ce3da70b43 Initial load
duke
parents:
diff changeset
   105
     */
90ce3da70b43 Initial load
duke
parents:
diff changeset
   106
    private final Comparator<? super E> comparator;
90ce3da70b43 Initial load
duke
parents:
diff changeset
   107
90ce3da70b43 Initial load
duke
parents:
diff changeset
   108
    /**
90ce3da70b43 Initial load
duke
parents:
diff changeset
   109
     * The number of times this priority queue has been
90ce3da70b43 Initial load
duke
parents:
diff changeset
   110
     * <i>structurally modified</i>.  See AbstractList for gory details.
90ce3da70b43 Initial load
duke
parents:
diff changeset
   111
     */
90ce3da70b43 Initial load
duke
parents:
diff changeset
   112
    private transient int modCount = 0;
90ce3da70b43 Initial load
duke
parents:
diff changeset
   113
90ce3da70b43 Initial load
duke
parents:
diff changeset
   114
    /**
90ce3da70b43 Initial load
duke
parents:
diff changeset
   115
     * Creates a {@code PriorityQueue} with the default initial
90ce3da70b43 Initial load
duke
parents:
diff changeset
   116
     * capacity (11) that orders its elements according to their
90ce3da70b43 Initial load
duke
parents:
diff changeset
   117
     * {@linkplain Comparable natural ordering}.
90ce3da70b43 Initial load
duke
parents:
diff changeset
   118
     */
90ce3da70b43 Initial load
duke
parents:
diff changeset
   119
    public PriorityQueue() {
90ce3da70b43 Initial load
duke
parents:
diff changeset
   120
        this(DEFAULT_INITIAL_CAPACITY, null);
90ce3da70b43 Initial load
duke
parents:
diff changeset
   121
    }
90ce3da70b43 Initial load
duke
parents:
diff changeset
   122
90ce3da70b43 Initial load
duke
parents:
diff changeset
   123
    /**
90ce3da70b43 Initial load
duke
parents:
diff changeset
   124
     * Creates a {@code PriorityQueue} with the specified initial
90ce3da70b43 Initial load
duke
parents:
diff changeset
   125
     * capacity that orders its elements according to their
90ce3da70b43 Initial load
duke
parents:
diff changeset
   126
     * {@linkplain Comparable natural ordering}.
90ce3da70b43 Initial load
duke
parents:
diff changeset
   127
     *
90ce3da70b43 Initial load
duke
parents:
diff changeset
   128
     * @param initialCapacity the initial capacity for this priority queue
90ce3da70b43 Initial load
duke
parents:
diff changeset
   129
     * @throws IllegalArgumentException if {@code initialCapacity} is less
90ce3da70b43 Initial load
duke
parents:
diff changeset
   130
     *         than 1
90ce3da70b43 Initial load
duke
parents:
diff changeset
   131
     */
90ce3da70b43 Initial load
duke
parents:
diff changeset
   132
    public PriorityQueue(int initialCapacity) {
90ce3da70b43 Initial load
duke
parents:
diff changeset
   133
        this(initialCapacity, null);
90ce3da70b43 Initial load
duke
parents:
diff changeset
   134
    }
90ce3da70b43 Initial load
duke
parents:
diff changeset
   135
90ce3da70b43 Initial load
duke
parents:
diff changeset
   136
    /**
90ce3da70b43 Initial load
duke
parents:
diff changeset
   137
     * Creates a {@code PriorityQueue} with the specified initial capacity
90ce3da70b43 Initial load
duke
parents:
diff changeset
   138
     * that orders its elements according to the specified comparator.
90ce3da70b43 Initial load
duke
parents:
diff changeset
   139
     *
90ce3da70b43 Initial load
duke
parents:
diff changeset
   140
     * @param  initialCapacity the initial capacity for this priority queue
90ce3da70b43 Initial load
duke
parents:
diff changeset
   141
     * @param  comparator the comparator that will be used to order this
90ce3da70b43 Initial load
duke
parents:
diff changeset
   142
     *         priority queue.  If {@code null}, the {@linkplain Comparable
90ce3da70b43 Initial load
duke
parents:
diff changeset
   143
     *         natural ordering} of the elements will be used.
90ce3da70b43 Initial load
duke
parents:
diff changeset
   144
     * @throws IllegalArgumentException if {@code initialCapacity} is
90ce3da70b43 Initial load
duke
parents:
diff changeset
   145
     *         less than 1
90ce3da70b43 Initial load
duke
parents:
diff changeset
   146
     */
90ce3da70b43 Initial load
duke
parents:
diff changeset
   147
    public PriorityQueue(int initialCapacity,
90ce3da70b43 Initial load
duke
parents:
diff changeset
   148
                         Comparator<? super E> comparator) {
90ce3da70b43 Initial load
duke
parents:
diff changeset
   149
        // Note: This restriction of at least one is not actually needed,
90ce3da70b43 Initial load
duke
parents:
diff changeset
   150
        // but continues for 1.5 compatibility
90ce3da70b43 Initial load
duke
parents:
diff changeset
   151
        if (initialCapacity < 1)
90ce3da70b43 Initial load
duke
parents:
diff changeset
   152
            throw new IllegalArgumentException();
90ce3da70b43 Initial load
duke
parents:
diff changeset
   153
        this.queue = new Object[initialCapacity];
90ce3da70b43 Initial load
duke
parents:
diff changeset
   154
        this.comparator = comparator;
90ce3da70b43 Initial load
duke
parents:
diff changeset
   155
    }
90ce3da70b43 Initial load
duke
parents:
diff changeset
   156
90ce3da70b43 Initial load
duke
parents:
diff changeset
   157
    /**
90ce3da70b43 Initial load
duke
parents:
diff changeset
   158
     * Creates a {@code PriorityQueue} containing the elements in the
90ce3da70b43 Initial load
duke
parents:
diff changeset
   159
     * specified collection.  If the specified collection is an instance of
90ce3da70b43 Initial load
duke
parents:
diff changeset
   160
     * a {@link SortedSet} or is another {@code PriorityQueue}, this
90ce3da70b43 Initial load
duke
parents:
diff changeset
   161
     * priority queue will be ordered according to the same ordering.
90ce3da70b43 Initial load
duke
parents:
diff changeset
   162
     * Otherwise, this priority queue will be ordered according to the
90ce3da70b43 Initial load
duke
parents:
diff changeset
   163
     * {@linkplain Comparable natural ordering} of its elements.
90ce3da70b43 Initial load
duke
parents:
diff changeset
   164
     *
90ce3da70b43 Initial load
duke
parents:
diff changeset
   165
     * @param  c the collection whose elements are to be placed
90ce3da70b43 Initial load
duke
parents:
diff changeset
   166
     *         into this priority queue
90ce3da70b43 Initial load
duke
parents:
diff changeset
   167
     * @throws ClassCastException if elements of the specified collection
90ce3da70b43 Initial load
duke
parents:
diff changeset
   168
     *         cannot be compared to one another according to the priority
90ce3da70b43 Initial load
duke
parents:
diff changeset
   169
     *         queue's ordering
90ce3da70b43 Initial load
duke
parents:
diff changeset
   170
     * @throws NullPointerException if the specified collection or any
90ce3da70b43 Initial load
duke
parents:
diff changeset
   171
     *         of its elements are null
90ce3da70b43 Initial load
duke
parents:
diff changeset
   172
     */
5467
0cfa1c70b5ab 6950540: PriorityQueue(collection) should throw NPE if collection contains a null
martin
parents: 5466
diff changeset
   173
    @SuppressWarnings("unchecked")
2
90ce3da70b43 Initial load
duke
parents:
diff changeset
   174
    public PriorityQueue(Collection<? extends E> c) {
5467
0cfa1c70b5ab 6950540: PriorityQueue(collection) should throw NPE if collection contains a null
martin
parents: 5466
diff changeset
   175
        if (c instanceof SortedSet<?>) {
0cfa1c70b5ab 6950540: PriorityQueue(collection) should throw NPE if collection contains a null
martin
parents: 5466
diff changeset
   176
            SortedSet<? extends E> ss = (SortedSet<? extends E>) c;
0cfa1c70b5ab 6950540: PriorityQueue(collection) should throw NPE if collection contains a null
martin
parents: 5466
diff changeset
   177
            this.comparator = (Comparator<? super E>) ss.comparator();
0cfa1c70b5ab 6950540: PriorityQueue(collection) should throw NPE if collection contains a null
martin
parents: 5466
diff changeset
   178
            initElementsFromCollection(ss);
0cfa1c70b5ab 6950540: PriorityQueue(collection) should throw NPE if collection contains a null
martin
parents: 5466
diff changeset
   179
        }
0cfa1c70b5ab 6950540: PriorityQueue(collection) should throw NPE if collection contains a null
martin
parents: 5466
diff changeset
   180
        else if (c instanceof PriorityQueue<?>) {
0cfa1c70b5ab 6950540: PriorityQueue(collection) should throw NPE if collection contains a null
martin
parents: 5466
diff changeset
   181
            PriorityQueue<? extends E> pq = (PriorityQueue<? extends E>) c;
0cfa1c70b5ab 6950540: PriorityQueue(collection) should throw NPE if collection contains a null
martin
parents: 5466
diff changeset
   182
            this.comparator = (Comparator<? super E>) pq.comparator();
0cfa1c70b5ab 6950540: PriorityQueue(collection) should throw NPE if collection contains a null
martin
parents: 5466
diff changeset
   183
            initFromPriorityQueue(pq);
0cfa1c70b5ab 6950540: PriorityQueue(collection) should throw NPE if collection contains a null
martin
parents: 5466
diff changeset
   184
        }
2
90ce3da70b43 Initial load
duke
parents:
diff changeset
   185
        else {
5467
0cfa1c70b5ab 6950540: PriorityQueue(collection) should throw NPE if collection contains a null
martin
parents: 5466
diff changeset
   186
            this.comparator = null;
0cfa1c70b5ab 6950540: PriorityQueue(collection) should throw NPE if collection contains a null
martin
parents: 5466
diff changeset
   187
            initFromCollection(c);
2
90ce3da70b43 Initial load
duke
parents:
diff changeset
   188
        }
90ce3da70b43 Initial load
duke
parents:
diff changeset
   189
    }
90ce3da70b43 Initial load
duke
parents:
diff changeset
   190
90ce3da70b43 Initial load
duke
parents:
diff changeset
   191
    /**
90ce3da70b43 Initial load
duke
parents:
diff changeset
   192
     * Creates a {@code PriorityQueue} containing the elements in the
90ce3da70b43 Initial load
duke
parents:
diff changeset
   193
     * specified priority queue.  This priority queue will be
90ce3da70b43 Initial load
duke
parents:
diff changeset
   194
     * ordered according to the same ordering as the given priority
90ce3da70b43 Initial load
duke
parents:
diff changeset
   195
     * queue.
90ce3da70b43 Initial load
duke
parents:
diff changeset
   196
     *
90ce3da70b43 Initial load
duke
parents:
diff changeset
   197
     * @param  c the priority queue whose elements are to be placed
90ce3da70b43 Initial load
duke
parents:
diff changeset
   198
     *         into this priority queue
90ce3da70b43 Initial load
duke
parents:
diff changeset
   199
     * @throws ClassCastException if elements of {@code c} cannot be
90ce3da70b43 Initial load
duke
parents:
diff changeset
   200
     *         compared to one another according to {@code c}'s
90ce3da70b43 Initial load
duke
parents:
diff changeset
   201
     *         ordering
90ce3da70b43 Initial load
duke
parents:
diff changeset
   202
     * @throws NullPointerException if the specified priority queue or any
90ce3da70b43 Initial load
duke
parents:
diff changeset
   203
     *         of its elements are null
90ce3da70b43 Initial load
duke
parents:
diff changeset
   204
     */
5467
0cfa1c70b5ab 6950540: PriorityQueue(collection) should throw NPE if collection contains a null
martin
parents: 5466
diff changeset
   205
    @SuppressWarnings("unchecked")
2
90ce3da70b43 Initial load
duke
parents:
diff changeset
   206
    public PriorityQueue(PriorityQueue<? extends E> c) {
5467
0cfa1c70b5ab 6950540: PriorityQueue(collection) should throw NPE if collection contains a null
martin
parents: 5466
diff changeset
   207
        this.comparator = (Comparator<? super E>) c.comparator();
0cfa1c70b5ab 6950540: PriorityQueue(collection) should throw NPE if collection contains a null
martin
parents: 5466
diff changeset
   208
        initFromPriorityQueue(c);
2
90ce3da70b43 Initial load
duke
parents:
diff changeset
   209
    }
90ce3da70b43 Initial load
duke
parents:
diff changeset
   210
90ce3da70b43 Initial load
duke
parents:
diff changeset
   211
    /**
90ce3da70b43 Initial load
duke
parents:
diff changeset
   212
     * Creates a {@code PriorityQueue} containing the elements in the
90ce3da70b43 Initial load
duke
parents:
diff changeset
   213
     * specified sorted set.   This priority queue will be ordered
90ce3da70b43 Initial load
duke
parents:
diff changeset
   214
     * according to the same ordering as the given sorted set.
90ce3da70b43 Initial load
duke
parents:
diff changeset
   215
     *
90ce3da70b43 Initial load
duke
parents:
diff changeset
   216
     * @param  c the sorted set whose elements are to be placed
90ce3da70b43 Initial load
duke
parents:
diff changeset
   217
     *         into this priority queue
90ce3da70b43 Initial load
duke
parents:
diff changeset
   218
     * @throws ClassCastException if elements of the specified sorted
90ce3da70b43 Initial load
duke
parents:
diff changeset
   219
     *         set cannot be compared to one another according to the
90ce3da70b43 Initial load
duke
parents:
diff changeset
   220
     *         sorted set's ordering
90ce3da70b43 Initial load
duke
parents:
diff changeset
   221
     * @throws NullPointerException if the specified sorted set or any
90ce3da70b43 Initial load
duke
parents:
diff changeset
   222
     *         of its elements are null
90ce3da70b43 Initial load
duke
parents:
diff changeset
   223
     */
5467
0cfa1c70b5ab 6950540: PriorityQueue(collection) should throw NPE if collection contains a null
martin
parents: 5466
diff changeset
   224
    @SuppressWarnings("unchecked")
2
90ce3da70b43 Initial load
duke
parents:
diff changeset
   225
    public PriorityQueue(SortedSet<? extends E> c) {
5467
0cfa1c70b5ab 6950540: PriorityQueue(collection) should throw NPE if collection contains a null
martin
parents: 5466
diff changeset
   226
        this.comparator = (Comparator<? super E>) c.comparator();
0cfa1c70b5ab 6950540: PriorityQueue(collection) should throw NPE if collection contains a null
martin
parents: 5466
diff changeset
   227
        initElementsFromCollection(c);
0cfa1c70b5ab 6950540: PriorityQueue(collection) should throw NPE if collection contains a null
martin
parents: 5466
diff changeset
   228
    }
0cfa1c70b5ab 6950540: PriorityQueue(collection) should throw NPE if collection contains a null
martin
parents: 5466
diff changeset
   229
0cfa1c70b5ab 6950540: PriorityQueue(collection) should throw NPE if collection contains a null
martin
parents: 5466
diff changeset
   230
    private void initFromPriorityQueue(PriorityQueue<? extends E> c) {
0cfa1c70b5ab 6950540: PriorityQueue(collection) should throw NPE if collection contains a null
martin
parents: 5466
diff changeset
   231
        if (c.getClass() == PriorityQueue.class) {
0cfa1c70b5ab 6950540: PriorityQueue(collection) should throw NPE if collection contains a null
martin
parents: 5466
diff changeset
   232
            this.queue = c.toArray();
0cfa1c70b5ab 6950540: PriorityQueue(collection) should throw NPE if collection contains a null
martin
parents: 5466
diff changeset
   233
            this.size = c.size();
0cfa1c70b5ab 6950540: PriorityQueue(collection) should throw NPE if collection contains a null
martin
parents: 5466
diff changeset
   234
        } else {
0cfa1c70b5ab 6950540: PriorityQueue(collection) should throw NPE if collection contains a null
martin
parents: 5466
diff changeset
   235
            initFromCollection(c);
0cfa1c70b5ab 6950540: PriorityQueue(collection) should throw NPE if collection contains a null
martin
parents: 5466
diff changeset
   236
        }
0cfa1c70b5ab 6950540: PriorityQueue(collection) should throw NPE if collection contains a null
martin
parents: 5466
diff changeset
   237
    }
0cfa1c70b5ab 6950540: PriorityQueue(collection) should throw NPE if collection contains a null
martin
parents: 5466
diff changeset
   238
0cfa1c70b5ab 6950540: PriorityQueue(collection) should throw NPE if collection contains a null
martin
parents: 5466
diff changeset
   239
    private void initElementsFromCollection(Collection<? extends E> c) {
0cfa1c70b5ab 6950540: PriorityQueue(collection) should throw NPE if collection contains a null
martin
parents: 5466
diff changeset
   240
        Object[] a = c.toArray();
0cfa1c70b5ab 6950540: PriorityQueue(collection) should throw NPE if collection contains a null
martin
parents: 5466
diff changeset
   241
        // If c.toArray incorrectly doesn't return Object[], copy it.
0cfa1c70b5ab 6950540: PriorityQueue(collection) should throw NPE if collection contains a null
martin
parents: 5466
diff changeset
   242
        if (a.getClass() != Object[].class)
0cfa1c70b5ab 6950540: PriorityQueue(collection) should throw NPE if collection contains a null
martin
parents: 5466
diff changeset
   243
            a = Arrays.copyOf(a, a.length, Object[].class);
0cfa1c70b5ab 6950540: PriorityQueue(collection) should throw NPE if collection contains a null
martin
parents: 5466
diff changeset
   244
        int len = a.length;
0cfa1c70b5ab 6950540: PriorityQueue(collection) should throw NPE if collection contains a null
martin
parents: 5466
diff changeset
   245
        if (len == 1 || this.comparator != null)
0cfa1c70b5ab 6950540: PriorityQueue(collection) should throw NPE if collection contains a null
martin
parents: 5466
diff changeset
   246
            for (int i = 0; i < len; i++)
0cfa1c70b5ab 6950540: PriorityQueue(collection) should throw NPE if collection contains a null
martin
parents: 5466
diff changeset
   247
                if (a[i] == null)
0cfa1c70b5ab 6950540: PriorityQueue(collection) should throw NPE if collection contains a null
martin
parents: 5466
diff changeset
   248
                    throw new NullPointerException();
0cfa1c70b5ab 6950540: PriorityQueue(collection) should throw NPE if collection contains a null
martin
parents: 5466
diff changeset
   249
        this.queue = a;
0cfa1c70b5ab 6950540: PriorityQueue(collection) should throw NPE if collection contains a null
martin
parents: 5466
diff changeset
   250
        this.size = a.length;
2
90ce3da70b43 Initial load
duke
parents:
diff changeset
   251
    }
90ce3da70b43 Initial load
duke
parents:
diff changeset
   252
90ce3da70b43 Initial load
duke
parents:
diff changeset
   253
    /**
90ce3da70b43 Initial load
duke
parents:
diff changeset
   254
     * Initializes queue array with elements from the given Collection.
90ce3da70b43 Initial load
duke
parents:
diff changeset
   255
     *
90ce3da70b43 Initial load
duke
parents:
diff changeset
   256
     * @param c the collection
90ce3da70b43 Initial load
duke
parents:
diff changeset
   257
     */
90ce3da70b43 Initial load
duke
parents:
diff changeset
   258
    private void initFromCollection(Collection<? extends E> c) {
5467
0cfa1c70b5ab 6950540: PriorityQueue(collection) should throw NPE if collection contains a null
martin
parents: 5466
diff changeset
   259
        initElementsFromCollection(c);
0cfa1c70b5ab 6950540: PriorityQueue(collection) should throw NPE if collection contains a null
martin
parents: 5466
diff changeset
   260
        heapify();
2
90ce3da70b43 Initial load
duke
parents:
diff changeset
   261
    }
90ce3da70b43 Initial load
duke
parents:
diff changeset
   262
90ce3da70b43 Initial load
duke
parents:
diff changeset
   263
    /**
5466
f130bb07764b 6933217: Huge arrays handled poorly in core libraries
martin
parents: 2
diff changeset
   264
     * The maximum size of array to allocate.
f130bb07764b 6933217: Huge arrays handled poorly in core libraries
martin
parents: 2
diff changeset
   265
     * Some VMs reserve some header words in an array.
f130bb07764b 6933217: Huge arrays handled poorly in core libraries
martin
parents: 2
diff changeset
   266
     * Attempts to allocate larger arrays may result in
f130bb07764b 6933217: Huge arrays handled poorly in core libraries
martin
parents: 2
diff changeset
   267
     * OutOfMemoryError: Requested array size exceeds VM limit
f130bb07764b 6933217: Huge arrays handled poorly in core libraries
martin
parents: 2
diff changeset
   268
     */
f130bb07764b 6933217: Huge arrays handled poorly in core libraries
martin
parents: 2
diff changeset
   269
    private static final int MAX_ARRAY_SIZE = Integer.MAX_VALUE - 8;
f130bb07764b 6933217: Huge arrays handled poorly in core libraries
martin
parents: 2
diff changeset
   270
f130bb07764b 6933217: Huge arrays handled poorly in core libraries
martin
parents: 2
diff changeset
   271
    /**
2
90ce3da70b43 Initial load
duke
parents:
diff changeset
   272
     * Increases the capacity of the array.
90ce3da70b43 Initial load
duke
parents:
diff changeset
   273
     *
90ce3da70b43 Initial load
duke
parents:
diff changeset
   274
     * @param minCapacity the desired minimum capacity
90ce3da70b43 Initial load
duke
parents:
diff changeset
   275
     */
90ce3da70b43 Initial load
duke
parents:
diff changeset
   276
    private void grow(int minCapacity) {
90ce3da70b43 Initial load
duke
parents:
diff changeset
   277
        int oldCapacity = queue.length;
90ce3da70b43 Initial load
duke
parents:
diff changeset
   278
        // Double size if small; else grow by 50%
5466
f130bb07764b 6933217: Huge arrays handled poorly in core libraries
martin
parents: 2
diff changeset
   279
        int newCapacity = oldCapacity + ((oldCapacity < 64) ?
f130bb07764b 6933217: Huge arrays handled poorly in core libraries
martin
parents: 2
diff changeset
   280
                                         (oldCapacity + 2) :
f130bb07764b 6933217: Huge arrays handled poorly in core libraries
martin
parents: 2
diff changeset
   281
                                         (oldCapacity >> 1));
f130bb07764b 6933217: Huge arrays handled poorly in core libraries
martin
parents: 2
diff changeset
   282
        // overflow-conscious code
f130bb07764b 6933217: Huge arrays handled poorly in core libraries
martin
parents: 2
diff changeset
   283
        if (newCapacity - MAX_ARRAY_SIZE > 0)
f130bb07764b 6933217: Huge arrays handled poorly in core libraries
martin
parents: 2
diff changeset
   284
            newCapacity = hugeCapacity(minCapacity);
2
90ce3da70b43 Initial load
duke
parents:
diff changeset
   285
        queue = Arrays.copyOf(queue, newCapacity);
90ce3da70b43 Initial load
duke
parents:
diff changeset
   286
    }
90ce3da70b43 Initial load
duke
parents:
diff changeset
   287
5466
f130bb07764b 6933217: Huge arrays handled poorly in core libraries
martin
parents: 2
diff changeset
   288
    private static int hugeCapacity(int minCapacity) {
f130bb07764b 6933217: Huge arrays handled poorly in core libraries
martin
parents: 2
diff changeset
   289
        if (minCapacity < 0) // overflow
f130bb07764b 6933217: Huge arrays handled poorly in core libraries
martin
parents: 2
diff changeset
   290
            throw new OutOfMemoryError();
f130bb07764b 6933217: Huge arrays handled poorly in core libraries
martin
parents: 2
diff changeset
   291
        return (minCapacity > MAX_ARRAY_SIZE) ?
f130bb07764b 6933217: Huge arrays handled poorly in core libraries
martin
parents: 2
diff changeset
   292
            Integer.MAX_VALUE :
f130bb07764b 6933217: Huge arrays handled poorly in core libraries
martin
parents: 2
diff changeset
   293
            MAX_ARRAY_SIZE;
f130bb07764b 6933217: Huge arrays handled poorly in core libraries
martin
parents: 2
diff changeset
   294
    }
f130bb07764b 6933217: Huge arrays handled poorly in core libraries
martin
parents: 2
diff changeset
   295
2
90ce3da70b43 Initial load
duke
parents:
diff changeset
   296
    /**
90ce3da70b43 Initial load
duke
parents:
diff changeset
   297
     * Inserts the specified element into this priority queue.
90ce3da70b43 Initial load
duke
parents:
diff changeset
   298
     *
90ce3da70b43 Initial load
duke
parents:
diff changeset
   299
     * @return {@code true} (as specified by {@link Collection#add})
90ce3da70b43 Initial load
duke
parents:
diff changeset
   300
     * @throws ClassCastException if the specified element cannot be
90ce3da70b43 Initial load
duke
parents:
diff changeset
   301
     *         compared with elements currently in this priority queue
90ce3da70b43 Initial load
duke
parents:
diff changeset
   302
     *         according to the priority queue's ordering
90ce3da70b43 Initial load
duke
parents:
diff changeset
   303
     * @throws NullPointerException if the specified element is null
90ce3da70b43 Initial load
duke
parents:
diff changeset
   304
     */
90ce3da70b43 Initial load
duke
parents:
diff changeset
   305
    public boolean add(E e) {
90ce3da70b43 Initial load
duke
parents:
diff changeset
   306
        return offer(e);
90ce3da70b43 Initial load
duke
parents:
diff changeset
   307
    }
90ce3da70b43 Initial load
duke
parents:
diff changeset
   308
90ce3da70b43 Initial load
duke
parents:
diff changeset
   309
    /**
90ce3da70b43 Initial load
duke
parents:
diff changeset
   310
     * Inserts the specified element into this priority queue.
90ce3da70b43 Initial load
duke
parents:
diff changeset
   311
     *
90ce3da70b43 Initial load
duke
parents:
diff changeset
   312
     * @return {@code true} (as specified by {@link Queue#offer})
90ce3da70b43 Initial load
duke
parents:
diff changeset
   313
     * @throws ClassCastException if the specified element cannot be
90ce3da70b43 Initial load
duke
parents:
diff changeset
   314
     *         compared with elements currently in this priority queue
90ce3da70b43 Initial load
duke
parents:
diff changeset
   315
     *         according to the priority queue's ordering
90ce3da70b43 Initial load
duke
parents:
diff changeset
   316
     * @throws NullPointerException if the specified element is null
90ce3da70b43 Initial load
duke
parents:
diff changeset
   317
     */
90ce3da70b43 Initial load
duke
parents:
diff changeset
   318
    public boolean offer(E e) {
90ce3da70b43 Initial load
duke
parents:
diff changeset
   319
        if (e == null)
90ce3da70b43 Initial load
duke
parents:
diff changeset
   320
            throw new NullPointerException();
90ce3da70b43 Initial load
duke
parents:
diff changeset
   321
        modCount++;
90ce3da70b43 Initial load
duke
parents:
diff changeset
   322
        int i = size;
90ce3da70b43 Initial load
duke
parents:
diff changeset
   323
        if (i >= queue.length)
90ce3da70b43 Initial load
duke
parents:
diff changeset
   324
            grow(i + 1);
90ce3da70b43 Initial load
duke
parents:
diff changeset
   325
        size = i + 1;
90ce3da70b43 Initial load
duke
parents:
diff changeset
   326
        if (i == 0)
90ce3da70b43 Initial load
duke
parents:
diff changeset
   327
            queue[0] = e;
90ce3da70b43 Initial load
duke
parents:
diff changeset
   328
        else
90ce3da70b43 Initial load
duke
parents:
diff changeset
   329
            siftUp(i, e);
90ce3da70b43 Initial load
duke
parents:
diff changeset
   330
        return true;
90ce3da70b43 Initial load
duke
parents:
diff changeset
   331
    }
90ce3da70b43 Initial load
duke
parents:
diff changeset
   332
90ce3da70b43 Initial load
duke
parents:
diff changeset
   333
    public E peek() {
90ce3da70b43 Initial load
duke
parents:
diff changeset
   334
        if (size == 0)
90ce3da70b43 Initial load
duke
parents:
diff changeset
   335
            return null;
90ce3da70b43 Initial load
duke
parents:
diff changeset
   336
        return (E) queue[0];
90ce3da70b43 Initial load
duke
parents:
diff changeset
   337
    }
90ce3da70b43 Initial load
duke
parents:
diff changeset
   338
90ce3da70b43 Initial load
duke
parents:
diff changeset
   339
    private int indexOf(Object o) {
90ce3da70b43 Initial load
duke
parents:
diff changeset
   340
        if (o != null) {
90ce3da70b43 Initial load
duke
parents:
diff changeset
   341
            for (int i = 0; i < size; i++)
90ce3da70b43 Initial load
duke
parents:
diff changeset
   342
                if (o.equals(queue[i]))
90ce3da70b43 Initial load
duke
parents:
diff changeset
   343
                    return i;
90ce3da70b43 Initial load
duke
parents:
diff changeset
   344
        }
90ce3da70b43 Initial load
duke
parents:
diff changeset
   345
        return -1;
90ce3da70b43 Initial load
duke
parents:
diff changeset
   346
    }
90ce3da70b43 Initial load
duke
parents:
diff changeset
   347
90ce3da70b43 Initial load
duke
parents:
diff changeset
   348
    /**
90ce3da70b43 Initial load
duke
parents:
diff changeset
   349
     * Removes a single instance of the specified element from this queue,
90ce3da70b43 Initial load
duke
parents:
diff changeset
   350
     * if it is present.  More formally, removes an element {@code e} such
90ce3da70b43 Initial load
duke
parents:
diff changeset
   351
     * that {@code o.equals(e)}, if this queue contains one or more such
90ce3da70b43 Initial load
duke
parents:
diff changeset
   352
     * elements.  Returns {@code true} if and only if this queue contained
90ce3da70b43 Initial load
duke
parents:
diff changeset
   353
     * the specified element (or equivalently, if this queue changed as a
90ce3da70b43 Initial load
duke
parents:
diff changeset
   354
     * result of the call).
90ce3da70b43 Initial load
duke
parents:
diff changeset
   355
     *
90ce3da70b43 Initial load
duke
parents:
diff changeset
   356
     * @param o element to be removed from this queue, if present
90ce3da70b43 Initial load
duke
parents:
diff changeset
   357
     * @return {@code true} if this queue changed as a result of the call
90ce3da70b43 Initial load
duke
parents:
diff changeset
   358
     */
90ce3da70b43 Initial load
duke
parents:
diff changeset
   359
    public boolean remove(Object o) {
90ce3da70b43 Initial load
duke
parents:
diff changeset
   360
        int i = indexOf(o);
90ce3da70b43 Initial load
duke
parents:
diff changeset
   361
        if (i == -1)
90ce3da70b43 Initial load
duke
parents:
diff changeset
   362
            return false;
90ce3da70b43 Initial load
duke
parents:
diff changeset
   363
        else {
90ce3da70b43 Initial load
duke
parents:
diff changeset
   364
            removeAt(i);
90ce3da70b43 Initial load
duke
parents:
diff changeset
   365
            return true;
90ce3da70b43 Initial load
duke
parents:
diff changeset
   366
        }
90ce3da70b43 Initial load
duke
parents:
diff changeset
   367
    }
90ce3da70b43 Initial load
duke
parents:
diff changeset
   368
90ce3da70b43 Initial load
duke
parents:
diff changeset
   369
    /**
90ce3da70b43 Initial load
duke
parents:
diff changeset
   370
     * Version of remove using reference equality, not equals.
90ce3da70b43 Initial load
duke
parents:
diff changeset
   371
     * Needed by iterator.remove.
90ce3da70b43 Initial load
duke
parents:
diff changeset
   372
     *
90ce3da70b43 Initial load
duke
parents:
diff changeset
   373
     * @param o element to be removed from this queue, if present
90ce3da70b43 Initial load
duke
parents:
diff changeset
   374
     * @return {@code true} if removed
90ce3da70b43 Initial load
duke
parents:
diff changeset
   375
     */
90ce3da70b43 Initial load
duke
parents:
diff changeset
   376
    boolean removeEq(Object o) {
90ce3da70b43 Initial load
duke
parents:
diff changeset
   377
        for (int i = 0; i < size; i++) {
90ce3da70b43 Initial load
duke
parents:
diff changeset
   378
            if (o == queue[i]) {
90ce3da70b43 Initial load
duke
parents:
diff changeset
   379
                removeAt(i);
90ce3da70b43 Initial load
duke
parents:
diff changeset
   380
                return true;
90ce3da70b43 Initial load
duke
parents:
diff changeset
   381
            }
90ce3da70b43 Initial load
duke
parents:
diff changeset
   382
        }
90ce3da70b43 Initial load
duke
parents:
diff changeset
   383
        return false;
90ce3da70b43 Initial load
duke
parents:
diff changeset
   384
    }
90ce3da70b43 Initial load
duke
parents:
diff changeset
   385
90ce3da70b43 Initial load
duke
parents:
diff changeset
   386
    /**
90ce3da70b43 Initial load
duke
parents:
diff changeset
   387
     * Returns {@code true} if this queue contains the specified element.
90ce3da70b43 Initial load
duke
parents:
diff changeset
   388
     * More formally, returns {@code true} if and only if this queue contains
90ce3da70b43 Initial load
duke
parents:
diff changeset
   389
     * at least one element {@code e} such that {@code o.equals(e)}.
90ce3da70b43 Initial load
duke
parents:
diff changeset
   390
     *
90ce3da70b43 Initial load
duke
parents:
diff changeset
   391
     * @param o object to be checked for containment in this queue
90ce3da70b43 Initial load
duke
parents:
diff changeset
   392
     * @return {@code true} if this queue contains the specified element
90ce3da70b43 Initial load
duke
parents:
diff changeset
   393
     */
90ce3da70b43 Initial load
duke
parents:
diff changeset
   394
    public boolean contains(Object o) {
90ce3da70b43 Initial load
duke
parents:
diff changeset
   395
        return indexOf(o) != -1;
90ce3da70b43 Initial load
duke
parents:
diff changeset
   396
    }
90ce3da70b43 Initial load
duke
parents:
diff changeset
   397
90ce3da70b43 Initial load
duke
parents:
diff changeset
   398
    /**
90ce3da70b43 Initial load
duke
parents:
diff changeset
   399
     * Returns an array containing all of the elements in this queue.
90ce3da70b43 Initial load
duke
parents:
diff changeset
   400
     * The elements are in no particular order.
90ce3da70b43 Initial load
duke
parents:
diff changeset
   401
     *
90ce3da70b43 Initial load
duke
parents:
diff changeset
   402
     * <p>The returned array will be "safe" in that no references to it are
90ce3da70b43 Initial load
duke
parents:
diff changeset
   403
     * maintained by this queue.  (In other words, this method must allocate
90ce3da70b43 Initial load
duke
parents:
diff changeset
   404
     * a new array).  The caller is thus free to modify the returned array.
90ce3da70b43 Initial load
duke
parents:
diff changeset
   405
     *
90ce3da70b43 Initial load
duke
parents:
diff changeset
   406
     * <p>This method acts as bridge between array-based and collection-based
90ce3da70b43 Initial load
duke
parents:
diff changeset
   407
     * APIs.
90ce3da70b43 Initial load
duke
parents:
diff changeset
   408
     *
90ce3da70b43 Initial load
duke
parents:
diff changeset
   409
     * @return an array containing all of the elements in this queue
90ce3da70b43 Initial load
duke
parents:
diff changeset
   410
     */
90ce3da70b43 Initial load
duke
parents:
diff changeset
   411
    public Object[] toArray() {
90ce3da70b43 Initial load
duke
parents:
diff changeset
   412
        return Arrays.copyOf(queue, size);
90ce3da70b43 Initial load
duke
parents:
diff changeset
   413
    }
90ce3da70b43 Initial load
duke
parents:
diff changeset
   414
90ce3da70b43 Initial load
duke
parents:
diff changeset
   415
    /**
90ce3da70b43 Initial load
duke
parents:
diff changeset
   416
     * Returns an array containing all of the elements in this queue; the
90ce3da70b43 Initial load
duke
parents:
diff changeset
   417
     * runtime type of the returned array is that of the specified array.
90ce3da70b43 Initial load
duke
parents:
diff changeset
   418
     * The returned array elements are in no particular order.
90ce3da70b43 Initial load
duke
parents:
diff changeset
   419
     * If the queue fits in the specified array, it is returned therein.
90ce3da70b43 Initial load
duke
parents:
diff changeset
   420
     * Otherwise, a new array is allocated with the runtime type of the
90ce3da70b43 Initial load
duke
parents:
diff changeset
   421
     * specified array and the size of this queue.
90ce3da70b43 Initial load
duke
parents:
diff changeset
   422
     *
90ce3da70b43 Initial load
duke
parents:
diff changeset
   423
     * <p>If the queue fits in the specified array with room to spare
90ce3da70b43 Initial load
duke
parents:
diff changeset
   424
     * (i.e., the array has more elements than the queue), the element in
90ce3da70b43 Initial load
duke
parents:
diff changeset
   425
     * the array immediately following the end of the collection is set to
90ce3da70b43 Initial load
duke
parents:
diff changeset
   426
     * {@code null}.
90ce3da70b43 Initial load
duke
parents:
diff changeset
   427
     *
90ce3da70b43 Initial load
duke
parents:
diff changeset
   428
     * <p>Like the {@link #toArray()} method, this method acts as bridge between
90ce3da70b43 Initial load
duke
parents:
diff changeset
   429
     * array-based and collection-based APIs.  Further, this method allows
90ce3da70b43 Initial load
duke
parents:
diff changeset
   430
     * precise control over the runtime type of the output array, and may,
90ce3da70b43 Initial load
duke
parents:
diff changeset
   431
     * under certain circumstances, be used to save allocation costs.
90ce3da70b43 Initial load
duke
parents:
diff changeset
   432
     *
90ce3da70b43 Initial load
duke
parents:
diff changeset
   433
     * <p>Suppose <tt>x</tt> is a queue known to contain only strings.
90ce3da70b43 Initial load
duke
parents:
diff changeset
   434
     * The following code can be used to dump the queue into a newly
90ce3da70b43 Initial load
duke
parents:
diff changeset
   435
     * allocated array of <tt>String</tt>:
90ce3da70b43 Initial load
duke
parents:
diff changeset
   436
     *
90ce3da70b43 Initial load
duke
parents:
diff changeset
   437
     * <pre>
90ce3da70b43 Initial load
duke
parents:
diff changeset
   438
     *     String[] y = x.toArray(new String[0]);</pre>
90ce3da70b43 Initial load
duke
parents:
diff changeset
   439
     *
90ce3da70b43 Initial load
duke
parents:
diff changeset
   440
     * Note that <tt>toArray(new Object[0])</tt> is identical in function to
90ce3da70b43 Initial load
duke
parents:
diff changeset
   441
     * <tt>toArray()</tt>.
90ce3da70b43 Initial load
duke
parents:
diff changeset
   442
     *
90ce3da70b43 Initial load
duke
parents:
diff changeset
   443
     * @param a the array into which the elements of the queue are to
90ce3da70b43 Initial load
duke
parents:
diff changeset
   444
     *          be stored, if it is big enough; otherwise, a new array of the
90ce3da70b43 Initial load
duke
parents:
diff changeset
   445
     *          same runtime type is allocated for this purpose.
90ce3da70b43 Initial load
duke
parents:
diff changeset
   446
     * @return an array containing all of the elements in this queue
90ce3da70b43 Initial load
duke
parents:
diff changeset
   447
     * @throws ArrayStoreException if the runtime type of the specified array
90ce3da70b43 Initial load
duke
parents:
diff changeset
   448
     *         is not a supertype of the runtime type of every element in
90ce3da70b43 Initial load
duke
parents:
diff changeset
   449
     *         this queue
90ce3da70b43 Initial load
duke
parents:
diff changeset
   450
     * @throws NullPointerException if the specified array is null
90ce3da70b43 Initial load
duke
parents:
diff changeset
   451
     */
90ce3da70b43 Initial load
duke
parents:
diff changeset
   452
    public <T> T[] toArray(T[] a) {
90ce3da70b43 Initial load
duke
parents:
diff changeset
   453
        if (a.length < size)
90ce3da70b43 Initial load
duke
parents:
diff changeset
   454
            // Make a new array of a's runtime type, but my contents:
90ce3da70b43 Initial load
duke
parents:
diff changeset
   455
            return (T[]) Arrays.copyOf(queue, size, a.getClass());
90ce3da70b43 Initial load
duke
parents:
diff changeset
   456
        System.arraycopy(queue, 0, a, 0, size);
90ce3da70b43 Initial load
duke
parents:
diff changeset
   457
        if (a.length > size)
90ce3da70b43 Initial load
duke
parents:
diff changeset
   458
            a[size] = null;
90ce3da70b43 Initial load
duke
parents:
diff changeset
   459
        return a;
90ce3da70b43 Initial load
duke
parents:
diff changeset
   460
    }
90ce3da70b43 Initial load
duke
parents:
diff changeset
   461
90ce3da70b43 Initial load
duke
parents:
diff changeset
   462
    /**
90ce3da70b43 Initial load
duke
parents:
diff changeset
   463
     * Returns an iterator over the elements in this queue. The iterator
90ce3da70b43 Initial load
duke
parents:
diff changeset
   464
     * does not return the elements in any particular order.
90ce3da70b43 Initial load
duke
parents:
diff changeset
   465
     *
90ce3da70b43 Initial load
duke
parents:
diff changeset
   466
     * @return an iterator over the elements in this queue
90ce3da70b43 Initial load
duke
parents:
diff changeset
   467
     */
90ce3da70b43 Initial load
duke
parents:
diff changeset
   468
    public Iterator<E> iterator() {
90ce3da70b43 Initial load
duke
parents:
diff changeset
   469
        return new Itr();
90ce3da70b43 Initial load
duke
parents:
diff changeset
   470
    }
90ce3da70b43 Initial load
duke
parents:
diff changeset
   471
90ce3da70b43 Initial load
duke
parents:
diff changeset
   472
    private final class Itr implements Iterator<E> {
90ce3da70b43 Initial load
duke
parents:
diff changeset
   473
        /**
90ce3da70b43 Initial load
duke
parents:
diff changeset
   474
         * Index (into queue array) of element to be returned by
90ce3da70b43 Initial load
duke
parents:
diff changeset
   475
         * subsequent call to next.
90ce3da70b43 Initial load
duke
parents:
diff changeset
   476
         */
90ce3da70b43 Initial load
duke
parents:
diff changeset
   477
        private int cursor = 0;
90ce3da70b43 Initial load
duke
parents:
diff changeset
   478
90ce3da70b43 Initial load
duke
parents:
diff changeset
   479
        /**
90ce3da70b43 Initial load
duke
parents:
diff changeset
   480
         * Index of element returned by most recent call to next,
90ce3da70b43 Initial load
duke
parents:
diff changeset
   481
         * unless that element came from the forgetMeNot list.
90ce3da70b43 Initial load
duke
parents:
diff changeset
   482
         * Set to -1 if element is deleted by a call to remove.
90ce3da70b43 Initial load
duke
parents:
diff changeset
   483
         */
90ce3da70b43 Initial load
duke
parents:
diff changeset
   484
        private int lastRet = -1;
90ce3da70b43 Initial load
duke
parents:
diff changeset
   485
90ce3da70b43 Initial load
duke
parents:
diff changeset
   486
        /**
90ce3da70b43 Initial load
duke
parents:
diff changeset
   487
         * A queue of elements that were moved from the unvisited portion of
90ce3da70b43 Initial load
duke
parents:
diff changeset
   488
         * the heap into the visited portion as a result of "unlucky" element
90ce3da70b43 Initial load
duke
parents:
diff changeset
   489
         * removals during the iteration.  (Unlucky element removals are those
90ce3da70b43 Initial load
duke
parents:
diff changeset
   490
         * that require a siftup instead of a siftdown.)  We must visit all of
90ce3da70b43 Initial load
duke
parents:
diff changeset
   491
         * the elements in this list to complete the iteration.  We do this
90ce3da70b43 Initial load
duke
parents:
diff changeset
   492
         * after we've completed the "normal" iteration.
90ce3da70b43 Initial load
duke
parents:
diff changeset
   493
         *
90ce3da70b43 Initial load
duke
parents:
diff changeset
   494
         * We expect that most iterations, even those involving removals,
90ce3da70b43 Initial load
duke
parents:
diff changeset
   495
         * will not need to store elements in this field.
90ce3da70b43 Initial load
duke
parents:
diff changeset
   496
         */
90ce3da70b43 Initial load
duke
parents:
diff changeset
   497
        private ArrayDeque<E> forgetMeNot = null;
90ce3da70b43 Initial load
duke
parents:
diff changeset
   498
90ce3da70b43 Initial load
duke
parents:
diff changeset
   499
        /**
90ce3da70b43 Initial load
duke
parents:
diff changeset
   500
         * Element returned by the most recent call to next iff that
90ce3da70b43 Initial load
duke
parents:
diff changeset
   501
         * element was drawn from the forgetMeNot list.
90ce3da70b43 Initial load
duke
parents:
diff changeset
   502
         */
90ce3da70b43 Initial load
duke
parents:
diff changeset
   503
        private E lastRetElt = null;
90ce3da70b43 Initial load
duke
parents:
diff changeset
   504
90ce3da70b43 Initial load
duke
parents:
diff changeset
   505
        /**
90ce3da70b43 Initial load
duke
parents:
diff changeset
   506
         * The modCount value that the iterator believes that the backing
90ce3da70b43 Initial load
duke
parents:
diff changeset
   507
         * Queue should have.  If this expectation is violated, the iterator
90ce3da70b43 Initial load
duke
parents:
diff changeset
   508
         * has detected concurrent modification.
90ce3da70b43 Initial load
duke
parents:
diff changeset
   509
         */
90ce3da70b43 Initial load
duke
parents:
diff changeset
   510
        private int expectedModCount = modCount;
90ce3da70b43 Initial load
duke
parents:
diff changeset
   511
90ce3da70b43 Initial load
duke
parents:
diff changeset
   512
        public boolean hasNext() {
90ce3da70b43 Initial load
duke
parents:
diff changeset
   513
            return cursor < size ||
90ce3da70b43 Initial load
duke
parents:
diff changeset
   514
                (forgetMeNot != null && !forgetMeNot.isEmpty());
90ce3da70b43 Initial load
duke
parents:
diff changeset
   515
        }
90ce3da70b43 Initial load
duke
parents:
diff changeset
   516
90ce3da70b43 Initial load
duke
parents:
diff changeset
   517
        public E next() {
90ce3da70b43 Initial load
duke
parents:
diff changeset
   518
            if (expectedModCount != modCount)
90ce3da70b43 Initial load
duke
parents:
diff changeset
   519
                throw new ConcurrentModificationException();
90ce3da70b43 Initial load
duke
parents:
diff changeset
   520
            if (cursor < size)
90ce3da70b43 Initial load
duke
parents:
diff changeset
   521
                return (E) queue[lastRet = cursor++];
90ce3da70b43 Initial load
duke
parents:
diff changeset
   522
            if (forgetMeNot != null) {
90ce3da70b43 Initial load
duke
parents:
diff changeset
   523
                lastRet = -1;
90ce3da70b43 Initial load
duke
parents:
diff changeset
   524
                lastRetElt = forgetMeNot.poll();
90ce3da70b43 Initial load
duke
parents:
diff changeset
   525
                if (lastRetElt != null)
90ce3da70b43 Initial load
duke
parents:
diff changeset
   526
                    return lastRetElt;
90ce3da70b43 Initial load
duke
parents:
diff changeset
   527
            }
90ce3da70b43 Initial load
duke
parents:
diff changeset
   528
            throw new NoSuchElementException();
90ce3da70b43 Initial load
duke
parents:
diff changeset
   529
        }
90ce3da70b43 Initial load
duke
parents:
diff changeset
   530
90ce3da70b43 Initial load
duke
parents:
diff changeset
   531
        public void remove() {
90ce3da70b43 Initial load
duke
parents:
diff changeset
   532
            if (expectedModCount != modCount)
90ce3da70b43 Initial load
duke
parents:
diff changeset
   533
                throw new ConcurrentModificationException();
90ce3da70b43 Initial load
duke
parents:
diff changeset
   534
            if (lastRet != -1) {
90ce3da70b43 Initial load
duke
parents:
diff changeset
   535
                E moved = PriorityQueue.this.removeAt(lastRet);
90ce3da70b43 Initial load
duke
parents:
diff changeset
   536
                lastRet = -1;
90ce3da70b43 Initial load
duke
parents:
diff changeset
   537
                if (moved == null)
90ce3da70b43 Initial load
duke
parents:
diff changeset
   538
                    cursor--;
90ce3da70b43 Initial load
duke
parents:
diff changeset
   539
                else {
90ce3da70b43 Initial load
duke
parents:
diff changeset
   540
                    if (forgetMeNot == null)
90ce3da70b43 Initial load
duke
parents:
diff changeset
   541
                        forgetMeNot = new ArrayDeque<E>();
90ce3da70b43 Initial load
duke
parents:
diff changeset
   542
                    forgetMeNot.add(moved);
90ce3da70b43 Initial load
duke
parents:
diff changeset
   543
                }
90ce3da70b43 Initial load
duke
parents:
diff changeset
   544
            } else if (lastRetElt != null) {
90ce3da70b43 Initial load
duke
parents:
diff changeset
   545
                PriorityQueue.this.removeEq(lastRetElt);
90ce3da70b43 Initial load
duke
parents:
diff changeset
   546
                lastRetElt = null;
90ce3da70b43 Initial load
duke
parents:
diff changeset
   547
            } else {
90ce3da70b43 Initial load
duke
parents:
diff changeset
   548
                throw new IllegalStateException();
90ce3da70b43 Initial load
duke
parents:
diff changeset
   549
            }
90ce3da70b43 Initial load
duke
parents:
diff changeset
   550
            expectedModCount = modCount;
90ce3da70b43 Initial load
duke
parents:
diff changeset
   551
        }
90ce3da70b43 Initial load
duke
parents:
diff changeset
   552
    }
90ce3da70b43 Initial load
duke
parents:
diff changeset
   553
90ce3da70b43 Initial load
duke
parents:
diff changeset
   554
    public int size() {
90ce3da70b43 Initial load
duke
parents:
diff changeset
   555
        return size;
90ce3da70b43 Initial load
duke
parents:
diff changeset
   556
    }
90ce3da70b43 Initial load
duke
parents:
diff changeset
   557
90ce3da70b43 Initial load
duke
parents:
diff changeset
   558
    /**
90ce3da70b43 Initial load
duke
parents:
diff changeset
   559
     * Removes all of the elements from this priority queue.
90ce3da70b43 Initial load
duke
parents:
diff changeset
   560
     * The queue will be empty after this call returns.
90ce3da70b43 Initial load
duke
parents:
diff changeset
   561
     */
90ce3da70b43 Initial load
duke
parents:
diff changeset
   562
    public void clear() {
90ce3da70b43 Initial load
duke
parents:
diff changeset
   563
        modCount++;
90ce3da70b43 Initial load
duke
parents:
diff changeset
   564
        for (int i = 0; i < size; i++)
90ce3da70b43 Initial load
duke
parents:
diff changeset
   565
            queue[i] = null;
90ce3da70b43 Initial load
duke
parents:
diff changeset
   566
        size = 0;
90ce3da70b43 Initial load
duke
parents:
diff changeset
   567
    }
90ce3da70b43 Initial load
duke
parents:
diff changeset
   568
90ce3da70b43 Initial load
duke
parents:
diff changeset
   569
    public E poll() {
90ce3da70b43 Initial load
duke
parents:
diff changeset
   570
        if (size == 0)
90ce3da70b43 Initial load
duke
parents:
diff changeset
   571
            return null;
90ce3da70b43 Initial load
duke
parents:
diff changeset
   572
        int s = --size;
90ce3da70b43 Initial load
duke
parents:
diff changeset
   573
        modCount++;
90ce3da70b43 Initial load
duke
parents:
diff changeset
   574
        E result = (E) queue[0];
90ce3da70b43 Initial load
duke
parents:
diff changeset
   575
        E x = (E) queue[s];
90ce3da70b43 Initial load
duke
parents:
diff changeset
   576
        queue[s] = null;
90ce3da70b43 Initial load
duke
parents:
diff changeset
   577
        if (s != 0)
90ce3da70b43 Initial load
duke
parents:
diff changeset
   578
            siftDown(0, x);
90ce3da70b43 Initial load
duke
parents:
diff changeset
   579
        return result;
90ce3da70b43 Initial load
duke
parents:
diff changeset
   580
    }
90ce3da70b43 Initial load
duke
parents:
diff changeset
   581
90ce3da70b43 Initial load
duke
parents:
diff changeset
   582
    /**
90ce3da70b43 Initial load
duke
parents:
diff changeset
   583
     * Removes the ith element from queue.
90ce3da70b43 Initial load
duke
parents:
diff changeset
   584
     *
90ce3da70b43 Initial load
duke
parents:
diff changeset
   585
     * Normally this method leaves the elements at up to i-1,
90ce3da70b43 Initial load
duke
parents:
diff changeset
   586
     * inclusive, untouched.  Under these circumstances, it returns
90ce3da70b43 Initial load
duke
parents:
diff changeset
   587
     * null.  Occasionally, in order to maintain the heap invariant,
90ce3da70b43 Initial load
duke
parents:
diff changeset
   588
     * it must swap a later element of the list with one earlier than
90ce3da70b43 Initial load
duke
parents:
diff changeset
   589
     * i.  Under these circumstances, this method returns the element
90ce3da70b43 Initial load
duke
parents:
diff changeset
   590
     * that was previously at the end of the list and is now at some
90ce3da70b43 Initial load
duke
parents:
diff changeset
   591
     * position before i. This fact is used by iterator.remove so as to
90ce3da70b43 Initial load
duke
parents:
diff changeset
   592
     * avoid missing traversing elements.
90ce3da70b43 Initial load
duke
parents:
diff changeset
   593
     */
90ce3da70b43 Initial load
duke
parents:
diff changeset
   594
    private E removeAt(int i) {
90ce3da70b43 Initial load
duke
parents:
diff changeset
   595
        assert i >= 0 && i < size;
90ce3da70b43 Initial load
duke
parents:
diff changeset
   596
        modCount++;
90ce3da70b43 Initial load
duke
parents:
diff changeset
   597
        int s = --size;
90ce3da70b43 Initial load
duke
parents:
diff changeset
   598
        if (s == i) // removed last element
90ce3da70b43 Initial load
duke
parents:
diff changeset
   599
            queue[i] = null;
90ce3da70b43 Initial load
duke
parents:
diff changeset
   600
        else {
90ce3da70b43 Initial load
duke
parents:
diff changeset
   601
            E moved = (E) queue[s];
90ce3da70b43 Initial load
duke
parents:
diff changeset
   602
            queue[s] = null;
90ce3da70b43 Initial load
duke
parents:
diff changeset
   603
            siftDown(i, moved);
90ce3da70b43 Initial load
duke
parents:
diff changeset
   604
            if (queue[i] == moved) {
90ce3da70b43 Initial load
duke
parents:
diff changeset
   605
                siftUp(i, moved);
90ce3da70b43 Initial load
duke
parents:
diff changeset
   606
                if (queue[i] != moved)
90ce3da70b43 Initial load
duke
parents:
diff changeset
   607
                    return moved;
90ce3da70b43 Initial load
duke
parents:
diff changeset
   608
            }
90ce3da70b43 Initial load
duke
parents:
diff changeset
   609
        }
90ce3da70b43 Initial load
duke
parents:
diff changeset
   610
        return null;
90ce3da70b43 Initial load
duke
parents:
diff changeset
   611
    }
90ce3da70b43 Initial load
duke
parents:
diff changeset
   612
90ce3da70b43 Initial load
duke
parents:
diff changeset
   613
    /**
90ce3da70b43 Initial load
duke
parents:
diff changeset
   614
     * Inserts item x at position k, maintaining heap invariant by
90ce3da70b43 Initial load
duke
parents:
diff changeset
   615
     * promoting x up the tree until it is greater than or equal to
90ce3da70b43 Initial load
duke
parents:
diff changeset
   616
     * its parent, or is the root.
90ce3da70b43 Initial load
duke
parents:
diff changeset
   617
     *
90ce3da70b43 Initial load
duke
parents:
diff changeset
   618
     * To simplify and speed up coercions and comparisons. the
90ce3da70b43 Initial load
duke
parents:
diff changeset
   619
     * Comparable and Comparator versions are separated into different
90ce3da70b43 Initial load
duke
parents:
diff changeset
   620
     * methods that are otherwise identical. (Similarly for siftDown.)
90ce3da70b43 Initial load
duke
parents:
diff changeset
   621
     *
90ce3da70b43 Initial load
duke
parents:
diff changeset
   622
     * @param k the position to fill
90ce3da70b43 Initial load
duke
parents:
diff changeset
   623
     * @param x the item to insert
90ce3da70b43 Initial load
duke
parents:
diff changeset
   624
     */
90ce3da70b43 Initial load
duke
parents:
diff changeset
   625
    private void siftUp(int k, E x) {
90ce3da70b43 Initial load
duke
parents:
diff changeset
   626
        if (comparator != null)
90ce3da70b43 Initial load
duke
parents:
diff changeset
   627
            siftUpUsingComparator(k, x);
90ce3da70b43 Initial load
duke
parents:
diff changeset
   628
        else
90ce3da70b43 Initial load
duke
parents:
diff changeset
   629
            siftUpComparable(k, x);
90ce3da70b43 Initial load
duke
parents:
diff changeset
   630
    }
90ce3da70b43 Initial load
duke
parents:
diff changeset
   631
90ce3da70b43 Initial load
duke
parents:
diff changeset
   632
    private void siftUpComparable(int k, E x) {
90ce3da70b43 Initial load
duke
parents:
diff changeset
   633
        Comparable<? super E> key = (Comparable<? super E>) x;
90ce3da70b43 Initial load
duke
parents:
diff changeset
   634
        while (k > 0) {
90ce3da70b43 Initial load
duke
parents:
diff changeset
   635
            int parent = (k - 1) >>> 1;
90ce3da70b43 Initial load
duke
parents:
diff changeset
   636
            Object e = queue[parent];
90ce3da70b43 Initial load
duke
parents:
diff changeset
   637
            if (key.compareTo((E) e) >= 0)
90ce3da70b43 Initial load
duke
parents:
diff changeset
   638
                break;
90ce3da70b43 Initial load
duke
parents:
diff changeset
   639
            queue[k] = e;
90ce3da70b43 Initial load
duke
parents:
diff changeset
   640
            k = parent;
90ce3da70b43 Initial load
duke
parents:
diff changeset
   641
        }
90ce3da70b43 Initial load
duke
parents:
diff changeset
   642
        queue[k] = key;
90ce3da70b43 Initial load
duke
parents:
diff changeset
   643
    }
90ce3da70b43 Initial load
duke
parents:
diff changeset
   644
90ce3da70b43 Initial load
duke
parents:
diff changeset
   645
    private void siftUpUsingComparator(int k, E x) {
90ce3da70b43 Initial load
duke
parents:
diff changeset
   646
        while (k > 0) {
90ce3da70b43 Initial load
duke
parents:
diff changeset
   647
            int parent = (k - 1) >>> 1;
90ce3da70b43 Initial load
duke
parents:
diff changeset
   648
            Object e = queue[parent];
90ce3da70b43 Initial load
duke
parents:
diff changeset
   649
            if (comparator.compare(x, (E) e) >= 0)
90ce3da70b43 Initial load
duke
parents:
diff changeset
   650
                break;
90ce3da70b43 Initial load
duke
parents:
diff changeset
   651
            queue[k] = e;
90ce3da70b43 Initial load
duke
parents:
diff changeset
   652
            k = parent;
90ce3da70b43 Initial load
duke
parents:
diff changeset
   653
        }
90ce3da70b43 Initial load
duke
parents:
diff changeset
   654
        queue[k] = x;
90ce3da70b43 Initial load
duke
parents:
diff changeset
   655
    }
90ce3da70b43 Initial load
duke
parents:
diff changeset
   656
90ce3da70b43 Initial load
duke
parents:
diff changeset
   657
    /**
90ce3da70b43 Initial load
duke
parents:
diff changeset
   658
     * Inserts item x at position k, maintaining heap invariant by
90ce3da70b43 Initial load
duke
parents:
diff changeset
   659
     * demoting x down the tree repeatedly until it is less than or
90ce3da70b43 Initial load
duke
parents:
diff changeset
   660
     * equal to its children or is a leaf.
90ce3da70b43 Initial load
duke
parents:
diff changeset
   661
     *
90ce3da70b43 Initial load
duke
parents:
diff changeset
   662
     * @param k the position to fill
90ce3da70b43 Initial load
duke
parents:
diff changeset
   663
     * @param x the item to insert
90ce3da70b43 Initial load
duke
parents:
diff changeset
   664
     */
90ce3da70b43 Initial load
duke
parents:
diff changeset
   665
    private void siftDown(int k, E x) {
90ce3da70b43 Initial load
duke
parents:
diff changeset
   666
        if (comparator != null)
90ce3da70b43 Initial load
duke
parents:
diff changeset
   667
            siftDownUsingComparator(k, x);
90ce3da70b43 Initial load
duke
parents:
diff changeset
   668
        else
90ce3da70b43 Initial load
duke
parents:
diff changeset
   669
            siftDownComparable(k, x);
90ce3da70b43 Initial load
duke
parents:
diff changeset
   670
    }
90ce3da70b43 Initial load
duke
parents:
diff changeset
   671
90ce3da70b43 Initial load
duke
parents:
diff changeset
   672
    private void siftDownComparable(int k, E x) {
90ce3da70b43 Initial load
duke
parents:
diff changeset
   673
        Comparable<? super E> key = (Comparable<? super E>)x;
90ce3da70b43 Initial load
duke
parents:
diff changeset
   674
        int half = size >>> 1;        // loop while a non-leaf
90ce3da70b43 Initial load
duke
parents:
diff changeset
   675
        while (k < half) {
90ce3da70b43 Initial load
duke
parents:
diff changeset
   676
            int child = (k << 1) + 1; // assume left child is least
90ce3da70b43 Initial load
duke
parents:
diff changeset
   677
            Object c = queue[child];
90ce3da70b43 Initial load
duke
parents:
diff changeset
   678
            int right = child + 1;
90ce3da70b43 Initial load
duke
parents:
diff changeset
   679
            if (right < size &&
90ce3da70b43 Initial load
duke
parents:
diff changeset
   680
                ((Comparable<? super E>) c).compareTo((E) queue[right]) > 0)
90ce3da70b43 Initial load
duke
parents:
diff changeset
   681
                c = queue[child = right];
90ce3da70b43 Initial load
duke
parents:
diff changeset
   682
            if (key.compareTo((E) c) <= 0)
90ce3da70b43 Initial load
duke
parents:
diff changeset
   683
                break;
90ce3da70b43 Initial load
duke
parents:
diff changeset
   684
            queue[k] = c;
90ce3da70b43 Initial load
duke
parents:
diff changeset
   685
            k = child;
90ce3da70b43 Initial load
duke
parents:
diff changeset
   686
        }
90ce3da70b43 Initial load
duke
parents:
diff changeset
   687
        queue[k] = key;
90ce3da70b43 Initial load
duke
parents:
diff changeset
   688
    }
90ce3da70b43 Initial load
duke
parents:
diff changeset
   689
90ce3da70b43 Initial load
duke
parents:
diff changeset
   690
    private void siftDownUsingComparator(int k, E x) {
90ce3da70b43 Initial load
duke
parents:
diff changeset
   691
        int half = size >>> 1;
90ce3da70b43 Initial load
duke
parents:
diff changeset
   692
        while (k < half) {
90ce3da70b43 Initial load
duke
parents:
diff changeset
   693
            int child = (k << 1) + 1;
90ce3da70b43 Initial load
duke
parents:
diff changeset
   694
            Object c = queue[child];
90ce3da70b43 Initial load
duke
parents:
diff changeset
   695
            int right = child + 1;
90ce3da70b43 Initial load
duke
parents:
diff changeset
   696
            if (right < size &&
90ce3da70b43 Initial load
duke
parents:
diff changeset
   697
                comparator.compare((E) c, (E) queue[right]) > 0)
90ce3da70b43 Initial load
duke
parents:
diff changeset
   698
                c = queue[child = right];
90ce3da70b43 Initial load
duke
parents:
diff changeset
   699
            if (comparator.compare(x, (E) c) <= 0)
90ce3da70b43 Initial load
duke
parents:
diff changeset
   700
                break;
90ce3da70b43 Initial load
duke
parents:
diff changeset
   701
            queue[k] = c;
90ce3da70b43 Initial load
duke
parents:
diff changeset
   702
            k = child;
90ce3da70b43 Initial load
duke
parents:
diff changeset
   703
        }
90ce3da70b43 Initial load
duke
parents:
diff changeset
   704
        queue[k] = x;
90ce3da70b43 Initial load
duke
parents:
diff changeset
   705
    }
90ce3da70b43 Initial load
duke
parents:
diff changeset
   706
90ce3da70b43 Initial load
duke
parents:
diff changeset
   707
    /**
90ce3da70b43 Initial load
duke
parents:
diff changeset
   708
     * Establishes the heap invariant (described above) in the entire tree,
90ce3da70b43 Initial load
duke
parents:
diff changeset
   709
     * assuming nothing about the order of the elements prior to the call.
90ce3da70b43 Initial load
duke
parents:
diff changeset
   710
     */
90ce3da70b43 Initial load
duke
parents:
diff changeset
   711
    private void heapify() {
90ce3da70b43 Initial load
duke
parents:
diff changeset
   712
        for (int i = (size >>> 1) - 1; i >= 0; i--)
90ce3da70b43 Initial load
duke
parents:
diff changeset
   713
            siftDown(i, (E) queue[i]);
90ce3da70b43 Initial load
duke
parents:
diff changeset
   714
    }
90ce3da70b43 Initial load
duke
parents:
diff changeset
   715
90ce3da70b43 Initial load
duke
parents:
diff changeset
   716
    /**
90ce3da70b43 Initial load
duke
parents:
diff changeset
   717
     * Returns the comparator used to order the elements in this
90ce3da70b43 Initial load
duke
parents:
diff changeset
   718
     * queue, or {@code null} if this queue is sorted according to
90ce3da70b43 Initial load
duke
parents:
diff changeset
   719
     * the {@linkplain Comparable natural ordering} of its elements.
90ce3da70b43 Initial load
duke
parents:
diff changeset
   720
     *
90ce3da70b43 Initial load
duke
parents:
diff changeset
   721
     * @return the comparator used to order this queue, or
90ce3da70b43 Initial load
duke
parents:
diff changeset
   722
     *         {@code null} if this queue is sorted according to the
90ce3da70b43 Initial load
duke
parents:
diff changeset
   723
     *         natural ordering of its elements
90ce3da70b43 Initial load
duke
parents:
diff changeset
   724
     */
90ce3da70b43 Initial load
duke
parents:
diff changeset
   725
    public Comparator<? super E> comparator() {
90ce3da70b43 Initial load
duke
parents:
diff changeset
   726
        return comparator;
90ce3da70b43 Initial load
duke
parents:
diff changeset
   727
    }
90ce3da70b43 Initial load
duke
parents:
diff changeset
   728
90ce3da70b43 Initial load
duke
parents:
diff changeset
   729
    /**
90ce3da70b43 Initial load
duke
parents:
diff changeset
   730
     * Saves the state of the instance to a stream (that
90ce3da70b43 Initial load
duke
parents:
diff changeset
   731
     * is, serializes it).
90ce3da70b43 Initial load
duke
parents:
diff changeset
   732
     *
90ce3da70b43 Initial load
duke
parents:
diff changeset
   733
     * @serialData The length of the array backing the instance is
90ce3da70b43 Initial load
duke
parents:
diff changeset
   734
     *             emitted (int), followed by all of its elements
90ce3da70b43 Initial load
duke
parents:
diff changeset
   735
     *             (each an {@code Object}) in the proper order.
90ce3da70b43 Initial load
duke
parents:
diff changeset
   736
     * @param s the stream
90ce3da70b43 Initial load
duke
parents:
diff changeset
   737
     */
90ce3da70b43 Initial load
duke
parents:
diff changeset
   738
    private void writeObject(java.io.ObjectOutputStream s)
90ce3da70b43 Initial load
duke
parents:
diff changeset
   739
        throws java.io.IOException{
90ce3da70b43 Initial load
duke
parents:
diff changeset
   740
        // Write out element count, and any hidden stuff
90ce3da70b43 Initial load
duke
parents:
diff changeset
   741
        s.defaultWriteObject();
90ce3da70b43 Initial load
duke
parents:
diff changeset
   742
90ce3da70b43 Initial load
duke
parents:
diff changeset
   743
        // Write out array length, for compatibility with 1.5 version
90ce3da70b43 Initial load
duke
parents:
diff changeset
   744
        s.writeInt(Math.max(2, size + 1));
90ce3da70b43 Initial load
duke
parents:
diff changeset
   745
90ce3da70b43 Initial load
duke
parents:
diff changeset
   746
        // Write out all elements in the "proper order".
90ce3da70b43 Initial load
duke
parents:
diff changeset
   747
        for (int i = 0; i < size; i++)
90ce3da70b43 Initial load
duke
parents:
diff changeset
   748
            s.writeObject(queue[i]);
90ce3da70b43 Initial load
duke
parents:
diff changeset
   749
    }
90ce3da70b43 Initial load
duke
parents:
diff changeset
   750
90ce3da70b43 Initial load
duke
parents:
diff changeset
   751
    /**
90ce3da70b43 Initial load
duke
parents:
diff changeset
   752
     * Reconstitutes the {@code PriorityQueue} instance from a stream
90ce3da70b43 Initial load
duke
parents:
diff changeset
   753
     * (that is, deserializes it).
90ce3da70b43 Initial load
duke
parents:
diff changeset
   754
     *
90ce3da70b43 Initial load
duke
parents:
diff changeset
   755
     * @param s the stream
90ce3da70b43 Initial load
duke
parents:
diff changeset
   756
     */
90ce3da70b43 Initial load
duke
parents:
diff changeset
   757
    private void readObject(java.io.ObjectInputStream s)
90ce3da70b43 Initial load
duke
parents:
diff changeset
   758
        throws java.io.IOException, ClassNotFoundException {
90ce3da70b43 Initial load
duke
parents:
diff changeset
   759
        // Read in size, and any hidden stuff
90ce3da70b43 Initial load
duke
parents:
diff changeset
   760
        s.defaultReadObject();
90ce3da70b43 Initial load
duke
parents:
diff changeset
   761
90ce3da70b43 Initial load
duke
parents:
diff changeset
   762
        // Read in (and discard) array length
90ce3da70b43 Initial load
duke
parents:
diff changeset
   763
        s.readInt();
90ce3da70b43 Initial load
duke
parents:
diff changeset
   764
90ce3da70b43 Initial load
duke
parents:
diff changeset
   765
        queue = new Object[size];
90ce3da70b43 Initial load
duke
parents:
diff changeset
   766
90ce3da70b43 Initial load
duke
parents:
diff changeset
   767
        // Read in all elements.
90ce3da70b43 Initial load
duke
parents:
diff changeset
   768
        for (int i = 0; i < size; i++)
90ce3da70b43 Initial load
duke
parents:
diff changeset
   769
            queue[i] = s.readObject();
90ce3da70b43 Initial load
duke
parents:
diff changeset
   770
90ce3da70b43 Initial load
duke
parents:
diff changeset
   771
        // Elements are guaranteed to be in "proper order", but the
90ce3da70b43 Initial load
duke
parents:
diff changeset
   772
        // spec has never explained what that might be.
90ce3da70b43 Initial load
duke
parents:
diff changeset
   773
        heapify();
90ce3da70b43 Initial load
duke
parents:
diff changeset
   774
    }
90ce3da70b43 Initial load
duke
parents:
diff changeset
   775
}