remove will remove the element from heap - Java Collection Framework

Java examples for Collection Framework:Heap

  1. HOME
  2. Java
  3. Collection Framework
  4. Heap

Description

remove will remove the element from heap

Demo Code


//package com.java2s;

public class Main {
    /**//from  w w  w . j ava 2 s.  co m
     * remove will remove the element from heap
     * 
     * @param heap
     *            is the heap from which the element is to be removed
     * @param heapSize
     *            is the size of the heap
     * @return
     */
    public static Comparable remove(Comparable[] heap, int heapSize) {
        Comparable root = heap[1];
        swap(heap, 1, heapSize);
        heapify(heap, 1, heapSize - 1);
        return root;
    }

    /**
     * swap will swap the elements in heap at index and index2
     * 
     * @param heap
     *            is the heap to be switched
     * @param index
     *            is the first element to be switched
     * @param index2
     *            is the second element to be switched
     */
    public static void swap(Comparable[] heap, int index, int index2) {
        Comparable temp = heap[index];
        heap[index] = heap[index2];
        heap[index2] = temp;
    }

    /**
     * heapifies starting from a given node
     * 
     * heapifying checks each child of the node and if it is greater than the
     * parent it swaps the two and heapifies the child
     * 
     * @param heap
     *            the heap to heapify in
     * @param index
     *            the index of the node to heapify
     * @param heapSize
     *            the size of the heap
     */
    public static void heapify(Comparable[] heap, int index, int heapSize) {
        int leftIndex = 2 * index;
        int rightIndex = 2 * index + 1;
        if (leftIndex <= heapSize) {
            if (heap[leftIndex].compareTo(heap[index]) > 0) {
                swap(heap, index, leftIndex);
                heapify(heap, leftIndex, heapSize);
            }
        }
        if (rightIndex <= heapSize) {
            if (heap[rightIndex].compareTo(heap[index]) > 0) {
                swap(heap, index, rightIndex);
                heapify(heap, rightIndex, heapSize);
            }
        }
    }
}

Related Tutorials