算法导论示例-PriorityQueue

原创 2006年06月04日 01:30:00
import java.util.EmptyStackException;

/**
 * Introduction to Algorithms, Second Edition 
 * 6.4 Priority queues
 * 
 * @author 土豆爸爸
 * 
 */
public class PriorityQueue<KeyType extends Comparable<KeyType>, T extends IPriorityQueueElement<KeyType>> {
    T[] array;

    int heapSize;

    /**
     * 构造函数
     * @param size 初始数组大小
     */
    @SuppressWarnings("unchecked")
    public PriorityQueue(int size) {
        array = (T[]) new IPriorityQueueElement[size];
    }

    /**
     * 获取当前heap中的最大值
     * 
     * @return 最大值
     */
    public T maximum() {
        return array[0];
    }

    /**
     * 获取当前heap中的最大值,并从heap中删除最大值
     * @return 最大值
     */
    public T extractMax() {
        if (heapSize < 1) {
            throw new EmptyStackException();
        }
        T max = array[0];
        array[0] = array[heapSize - 1];
        heapSize--;
        maxHeapify(0);
        return max;
    }

    /**
     * 插入一个元素
     * @param e
     */
    @SuppressWarnings("unchecked")
    public void insert(T e) {
        if (heapSize == array.length) {
            T[] newArray = (T[]) new IPriorityQueueElement[array.length * 2];
            System.arraycopy(array, 0, newArray, 0, array.length);
            array = newArray;
        }
        int i = heapSize++;
        array[i] = e;
        int p = parent(i); // 父结点索引
        while (i > 0 && array[p].getKey().compareTo(array[i].getKey()) < 0) {
            T temp = array[i];
            array[i] = array[p];
            array[p] = temp;
            i = p;
            p = parent(i);
        }
    }

    /**
     * 使数组的第i个元素按max heap规则重排
     * 
     * @param i
     *            元素索引
     */
    private void maxHeapify(int i) {
        int l = left(i);
        int r = right(i);
        int largest; // 当前结点/左子结点/右子结点中最大值的索引
        if (l < heapSize && array[l].getKey().compareTo(array[i].getKey()) > 0) {
            largest = l;
        } else {
            largest = i;
        }

        if (r < heapSize
                && array[r].getKey().compareTo(array[largest].getKey()) > 0) {
            largest = r;
        }

        if (largest != i) {
            // 如果最大值不是当前结点,进行交换
            T temp = array[i];
            array[i] = array[largest];
            array[largest] = temp;
            // 递归调用,直到当前结点比其子结点大
            maxHeapify(largest);
        }

    }

    /**
     * 计算结点索引为i的元素的父结点的索引
     * 
     * @param i
     *            当前索引
     * @return 父结点的索引
     */
    private int parent(int i) {
        return (i + 1) / 2 - 1;
    }

    /**
     * 计算结点索引为i的元素的左子结点的索引
     * 
     * @param i
     *            当前索引
     * @return 左子结点的索引
     */
    private int left(int i) {
        return 2 * i + 1;
    }

    /**
     * 计算结点索引为i的元素的右子结点的索引
     * 
     * @param i
     *            当前索引
     * @return 右子结点的索引
     */
    private int right(int i) {
        return 2 * i + 2;
    }
}

public interface IPriorityQueueElement<KeyType extends Comparable<KeyType>>{
    KeyType getKey();
}

import junit.framework.TestCase;

import junit.framework.TestCase;

public class PriorityQueueTest extends TestCase {
    class Element implements IPriorityQueueElement<Integer> {
        Integer key;

        public Element(Integer key) {
            this.key = key;
        }

        public Integer getKey() {
            return key;
        }
    }

    public void testMaximum() {
        PriorityQueue<Integer, Element> queue = new PriorityQueue<Integer, Element>(2);
        queue.insert(new Element(1));
        queue.insert(new Element(2));
        assertEquals(new Integer(2), queue.maximum().getKey());
        queue.insert(new Element(10));
        assertEquals(new Integer(10), queue.maximum().getKey());
        queue.insert(new Element(3));
        assertEquals(new Integer(10), queue.maximum().getKey());
    }
    
    public void testExtractMax() {
        PriorityQueue<Integer, Element> queue = new PriorityQueue<Integer, Element>(2);
        queue.insert(new Element(1));
        queue.insert(new Element(2));
        assertEquals(new Integer(2), queue.extractMax().getKey());
        assertEquals(new Integer(1), queue.maximum().getKey());
    }
}

算法导论22.2广度优先搜索 练习总结

22.2-1 请计算出在有向图 22-2(a) 上运行广度优先搜索算法后的 d 值和 π 值,这里假定结点 3 为算法所用的源结点。 ANSWER: 22.2-2 请计算出在图 22-3 所示无向...
  • chan15
  • chan15
  • 2015年11月14日 19:50
  • 2241

用Latex排版论文(4)《大连海事大学硕博学位论文Latex模板》中算法排版例子(添加中文输入和输出)

最近一直在写硕士毕业论文,在LaTeXStudio上发现了由@智博是小叮当编写的《大连海事大学硕博学位论文latex模板》,所以就开始尝试使用该模板来写毕业论文,有种前人种树后人乘凉的感觉。     ...
  • kbawyg
  • kbawyg
  • 2014年03月05日 00:50
  • 3546

算法导论第十一章----11.1.1-11.1.4

http://test.scripts.psu.edu/users/d/j/djh300/cmpsc465/notes-4985903869437/solutions-to-some-homework...
  • lihenair
  • lihenair
  • 2014年02月27日 18:20
  • 3042

链式哈希表(Hash Table)--算法导论示例

In computer science, a hash table is an associative array data structure that associates keys with v...
  • rockerman_shane
  • rockerman_shane
  • 2011年02月21日 20:06
  • 2137

算法导论与比赛常用算法示例_第1版

  • 2013年12月01日 22:44
  • 390KB
  • 下载

《算法导论(原书第2版)》(Thomas H.Cormen[美] 等 著,潘金贵 等 译)

  • 2018年01月16日 16:12
  • 46.89MB
  • 下载

算法导论答案第七章(第三版)详细讲解

  • 2018年01月15日 20:15
  • 202KB
  • 下载

算法导论答案第八章(第三版)详细讲解

  • 2018年01月15日 20:17
  • 172KB
  • 下载

算法导论中文第三版,非扫描,带书签

  • 2017年11月08日 08:03
  • 33.26MB
  • 下载

堆排序(C#,C++)算法导论

  • 2017年10月11日 10:50
  • 3.21MB
  • 下载
内容举报
返回顶部
收藏助手
不良信息举报
您举报文章:算法导论示例-PriorityQueue
举报原因:
原因补充:

(最多只允许输入30个字)