org.jheaps.array.DaryArrayHeap Maven / Gradle / Ivy
Show all versions of AptSpringProcessor Show documentation
/*
* (C) Copyright 2014-2016, by Dimitrios Michail
*
* JHeaps Library
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.jheaps.array;
import java.util.Comparator;
import org.jheaps.annotations.LinearTime;
/**
* An array based d-ary heap. The heap is sorted according to the
* {@linkplain Comparable natural ordering} of its keys, or by a
* {@link Comparator} provided at heap creation time, depending on which
* constructor is used.
*
*
* The implementation uses an array in order to store the elements and
* automatically maintains the size of the array much like a
* {@link java.util.Vector} does, providing amortized O(log_d(n)) time cost for
* the {@code insert} and amortized O(d log_d(n)) for the {@code deleteMin}
* operation. Operation {@code findMin}, is a worst-case O(1) operation. The
* bounds are worst-case if the user initializes the heap with a capacity larger
* or equal to the total number of elements that are going to be inserted into
* the heap.
*
*
* Constructing such a heap from an array of elements can be performed using the
* method {@link #heapify(int, Object[])} or
* {@link #heapify(int, Object[], Comparator)} in linear time.
*
*
* Note that the ordering maintained by a d-ary heap, like any heap, and whether
* or not an explicit comparator is provided, must be consistent with
* {@code equals} if this heap is to correctly implement the {@code Heap}
* interface. (See {@code Comparable} or {@code Comparator} for a precise
* definition of consistent with equals.) This is so because the
* {@code Heap} interface is defined in terms of the {@code equals} operation,
* but a d-ary heap performs all key comparisons using its {@code compareTo} (or
* {@code compare}) method, so two keys that are deemed equal by this method
* are, from the standpoint of the d-ary heap, equal. The behavior of a heap
* is well-defined even if its ordering is inconsistent with
* {@code equals}; it just fails to obey the general contract of the
* {@code Heap} interface.
*
*
* Note that this implementation is not synchronized. If
* multiple threads access a heap concurrently, and at least one of the threads
* modifies the heap structurally, it must be synchronized externally.
* (A structural modification is any operation that adds or deletes one or more
* elements or changing the key of some element.) This is typically accomplished
* by synchronizing on some object that naturally encapsulates the heap.
*
* @param
* the type of keys maintained by this heap
*
* @author Dimitrios Michail
*/
public class DaryArrayHeap extends AbstractArrayHeap {
private static final long serialVersionUID = 1L;
/**
* Default initial capacity of the heap.
*/
public static final int DEFAULT_HEAP_CAPACITY = 16;
/**
* Degree
*/
protected int d;
/**
* Constructs a new, empty heap, using the natural ordering of its keys.
*
*
* All keys inserted into the heap must implement the {@link Comparable}
* interface. Furthermore, all such keys must be mutually
* comparable: {@code k1.compareTo(k2)} must not throw a
* {@code ClassCastException} for any keys {@code k1} and {@code k2} in the
* heap. If the user attempts to put a key into the heap that violates this
* constraint (for example, the user attempts to put a string key into a
* heap whose keys are integers), the {@code insert(Object key)} call will
* throw a {@code ClassCastException}.
*
*
* The initial capacity of the heap is
* {@link DaryArrayHeap#DEFAULT_HEAP_CAPACITY} and adjusts automatically
* based on the sequence of insertions and deletions.
*
* @param d
* the number of children of each node in the d-ary heap
* @throws IllegalArgumentException
* in case the number of children per node are less than 2
*/
public DaryArrayHeap(int d) {
this(d, null, DEFAULT_HEAP_CAPACITY);
}
/**
* Constructs a new, empty heap, with a provided initial capacity using the
* natural ordering of its keys.
*
*
* All keys inserted into the heap must implement the {@link Comparable}
* interface. Furthermore, all such keys must be mutually
* comparable: {@code k1.compareTo(k2)} must not throw a
* {@code ClassCastException} for any keys {@code k1} and {@code k2} in the
* heap. If the user attempts to put a key into the heap that violates this
* constraint (for example, the user attempts to put a string key into a
* heap whose keys are integers), the {@code insert(Object key)} call will
* throw a {@code ClassCastException}.
*
*
* The initial capacity of the heap is provided by the user and is adjusted
* automatically based on the sequence of insertions and deletions. The
* capacity will never become smaller than the initial requested capacity.
*
* @param d
* the number of children of each node in the d-ary heap
* @param capacity
* the initial heap capacity
* @throws IllegalArgumentException
* in case the number of children per node are less than 2
*/
public DaryArrayHeap(int d, int capacity) {
this(d, null, capacity);
}
/**
* Constructs a new, empty heap, ordered according to the given comparator.
*
*
* All keys inserted into the heap must be mutually comparable by
* the given comparator: {@code comparator.compare(k1,
* k2)} must not throw a {@code ClassCastException} for any keys {@code k1}
* and {@code k2} in the heap. If the user attempts to put a key into the
* heap that violates this constraint, the {@code insert(Object key)} call
* will throw a {@code ClassCastException}.
*
*
* The initial capacity of the heap is
* {@link DaryArrayHeap#DEFAULT_HEAP_CAPACITY} and adjusts automatically
* based on the sequence of insertions and deletions.
*
* @param d
* the number of children of each node in the d-ary heap
* @param comparator
* the comparator that will be used to order this heap. If
* {@code null}, the {@linkplain Comparable natural ordering} of
* the keys will be used.
* @throws IllegalArgumentException
* in case the number of children per node are less than 2
*/
public DaryArrayHeap(int d, Comparator super K> comparator) {
this(d, comparator, DEFAULT_HEAP_CAPACITY);
}
/**
* Constructs a new, empty heap, with a provided initial capacity ordered
* according to the given comparator.
*
*
* All keys inserted into the heap must be mutually comparable by
* the given comparator: {@code comparator.compare(k1,
* k2)} must not throw a {@code ClassCastException} for any keys {@code k1}
* and {@code k2} in the heap. If the user attempts to put a key into the
* heap that violates this constraint, the {@code insert(Object key)} call
* will throw a {@code ClassCastException}.
*
*
* The initial capacity of the heap is provided by the user and is adjusted
* automatically based on the sequence of insertions and deletions. The
* capacity will never become smaller than the initial requested capacity.
*
* @param d
* the number of children of each node in the d-ary heap
* @param comparator
* the comparator that will be used to order this heap. If
* {@code null}, the {@linkplain Comparable natural ordering} of
* the keys will be used.
* @param capacity
* the initial heap capacity
* @throws IllegalArgumentException
* in case the number of children per node are less than 2
*/
public DaryArrayHeap(int d, Comparator super K> comparator, int capacity) {
super(comparator, capacity);
if (d < 2) {
throw new IllegalArgumentException("D-ary heaps must have at least 2 children per node");
}
this.d = d;
}
/**
* Create a heap from an array of elements. The elements of the array are
* not destroyed. The method has linear time complexity.
*
* @param
* the type of keys maintained by the heap
* @param d
* the number of children of the d-ary heap
* @param array
* an array of elements
* @return a d-ary heap
* @throws IllegalArgumentException
* in case the number of children per node are less than 2
* @throws IllegalArgumentException
* in case the array is null
*/
@LinearTime
public static DaryArrayHeap heapify(int d, K[] array) {
if (d < 2) {
throw new IllegalArgumentException("D-ary heaps must have at least 2 children per node");
}
if (array == null) {
throw new IllegalArgumentException("Array cannot be null");
}
if (array.length == 0) {
return new DaryArrayHeap(d);
}
DaryArrayHeap h = new DaryArrayHeap(d, array.length);
System.arraycopy(array, 0, h.array, 1, array.length);
h.size = array.length;
for (int i = array.length / d; i > 0; i--) {
h.fixdown(i);
}
return h;
}
/**
* Create a heap from an array of elements. The elements of the array are
* not destroyed. The method has linear time complexity.
*
* @param
* the type of keys maintained by the heap
* @param d
* the number of children of the d-ary heap
* @param array
* an array of elements
* @param comparator
* the comparator to use
* @return a d-ary heap
* @throws IllegalArgumentException
* in case the number of children per node are less than 2
* @throws IllegalArgumentException
* in case the array is null
*/
@LinearTime
public static DaryArrayHeap heapify(int d, K[] array, Comparator super K> comparator) {
if (d < 2) {
throw new IllegalArgumentException("D-ary heaps must have at least 2 children per node");
}
if (array == null) {
throw new IllegalArgumentException("Array cannot be null");
}
if (array.length == 0) {
return new DaryArrayHeap(d, comparator);
}
DaryArrayHeap h = new DaryArrayHeap(d, comparator, array.length);
System.arraycopy(array, 0, h.array, 1, array.length);
h.size = array.length;
for (int i = array.length / d; i > 0; i--) {
h.fixdownWithComparator(i);
}
return h;
}
/**
* Ensure that the array representation has the necessary capacity.
*
* @param capacity
* the requested capacity
*/
@Override
@SuppressWarnings("unchecked")
protected void ensureCapacity(int capacity) {
checkCapacity(capacity);
K[] newArray = (K[]) new Object[capacity + 1];
System.arraycopy(array, 1, newArray, 1, size);
array = newArray;
}
@Override
@SuppressWarnings("unchecked")
protected void fixup(int k) {
// assert k >= 1 && k <= size;
K key = array[k];
while (k > 1) {
int p = (k - 2) / d + 1;
if (((Comparable super K>) array[p]).compareTo(key) <= 0) {
break;
}
array[k] = array[p];
k = p;
}
array[k] = key;
}
@Override
protected void fixupWithComparator(int k) {
// assert k >= 1 && k <= size;
K key = array[k];
while (k > 1) {
int p = (k - 2) / d + 1;
if (comparator.compare(array[p], key) <= 0) {
break;
}
array[k] = array[p];
k = p;
}
array[k] = key;
}
@Override
@SuppressWarnings("unchecked")
protected void fixdown(int k) {
int c;
K key = array[k];
while ((c = d * (k - 1) + 2) <= size) {
int maxc = c;
for (int i = 1; i < d; i++) {
if (c + i <= size && ((Comparable super K>) array[maxc]).compareTo(array[c + i]) > 0) {
maxc = c + i;
}
}
if (((Comparable super K>) key).compareTo(array[maxc]) <= 0) {
break;
}
array[k] = array[maxc];
k = maxc;
}
array[k] = key;
}
@Override
protected void fixdownWithComparator(int k) {
int c;
K key = array[k];
while ((c = d * (k - 1) + 2) <= size) {
int maxc = c;
for (int i = 1; i < d; i++) {
if (c + i <= size && comparator.compare(array[maxc], array[c + i]) > 0) {
maxc = c + i;
}
}
if (comparator.compare(key, array[maxc]) <= 0) {
break;
}
array[k] = array[maxc];
k = maxc;
}
array[k] = key;
}
}