UNPKG

typescript-collections

Version:

A complete, fully tested data structure library written in TypeScript.

152 lines (151 loc) 5.25 kB
import * as collections from './util'; export default class Heap<T> { /** * Array used to store the elements of the heap. * @type {Array.<Object>} * @private */ private data; /** * Function used to compare elements. * @type {function(Object,Object):number} * @private */ private compare; /** * Creates an empty Heap. * @class * <p>A heap is a binary tree, where the nodes maintain the heap property: * each node is smaller than each of its children and therefore a MinHeap * This implementation uses an array to store elements.</p> * <p>If the inserted elements are custom objects a compare function must be provided, * at construction time, otherwise the <=, === and >= operators are * used to compare elements. Example:</p> * * <pre> * function compare(a, b) { * if (a is less than b by some ordering criterion) { * return -1; * } if (a is greater than b by the ordering criterion) { * return 1; * } * // a must be equal to b * return 0; * } * </pre> * * <p>If a Max-Heap is wanted (greater elements on top) you can a provide a * reverse compare function to accomplish that behavior. Example:</p> * * <pre> * function reverseCompare(a, b) { * if (a is less than b by some ordering criterion) { * return 1; * } if (a is greater than b by the ordering criterion) { * return -1; * } * // a must be equal to b * return 0; * } * </pre> * * @constructor * @param {function(Object,Object):number=} compareFunction optional * function used to compare two elements. Must return a negative integer, * zero, or a positive integer as the first argument is less than, equal to, * or greater than the second. */ constructor(compareFunction?: collections.ICompareFunction<T>); /** * Returns the index of the left child of the node at the given index. * @param {number} nodeIndex The index of the node to get the left child * for. * @return {number} The index of the left child. * @private */ private leftChildIndex(nodeIndex); /** * Returns the index of the right child of the node at the given index. * @param {number} nodeIndex The index of the node to get the right child * for. * @return {number} The index of the right child. * @private */ private rightChildIndex(nodeIndex); /** * Returns the index of the parent of the node at the given index. * @param {number} nodeIndex The index of the node to get the parent for. * @return {number} The index of the parent. * @private */ private parentIndex(nodeIndex); /** * Returns the index of the smaller child node (if it exists). * @param {number} leftChild left child index. * @param {number} rightChild right child index. * @return {number} the index with the minimum value or -1 if it doesn't * exists. * @private */ private minIndex(leftChild, rightChild); /** * Moves the node at the given index up to its proper place in the heap. * @param {number} index The index of the node to move up. * @private */ private siftUp(index); /** * Moves the node at the given index down to its proper place in the heap. * @param {number} nodeIndex The index of the node to move down. * @private */ private siftDown(nodeIndex); /** * Retrieves but does not remove the root element of this heap. * @return {*} The value at the root of the heap. Returns undefined if the * heap is empty. */ peek(): T | undefined; /** * Adds the given element into the heap. * @param {*} element the element. * @return true if the element was added or fals if it is undefined. */ add(element: T): boolean; /** * Retrieves and removes the root element of this heap. * @return {*} The value removed from the root of the heap. Returns * undefined if the heap is empty. */ removeRoot(): T | undefined; /** * Returns true if this heap contains the specified element. * @param {Object} element element to search for. * @return {boolean} true if this Heap contains the specified element, false * otherwise. */ contains(element: T): boolean; /** * Returns the number of elements in this heap. * @return {number} the number of elements in this heap. */ size(): number; /** * Checks if this heap is empty. * @return {boolean} true if and only if this heap contains no items; false * otherwise. */ isEmpty(): boolean; /** * Removes all of the elements from this heap. */ clear(): void; /** * Executes the provided function once for each element present in this heap in * no particular order. * @param {function(Object):*} callback function to execute, it is * invoked with one argument: the element value, to break the iteration you can * optionally return false. */ forEach(callback: collections.ILoopFunction<T>): void; }