基于堆的优先队列

队列接口:


public interface Queue<E> {
	int getSize();
	boolean isEmpty();
	void enqueue(E e);
	E dequeue();
	E getFront();
}

最大堆代码(基于完全二叉树):


public class MaxHeap<E extends Comparable<E>> {
	private Array<E> data;

	public MaxHeap(int capacity) {
		data = new Array<>(capacity);
	}

	public MaxHeap() {
		data = new Array<>();
	}

	public MaxHeap(E[] arr) {  
		data = new Array<>(arr);
		for (int i = parent(arr.length - 1); i >= 0; i--)
			siftDown(i);
	}  

	// 返回堆中的元素个数
	public int size() {
		return data.getSize();
	}

	// 返回一个布尔值,表示堆中是否为空
	public boolean isEmpty() {
		return data.isEmpty();
	}

	// 返回完全二叉树的数组表示中,一个索引所表示的元素的父亲节点的索引
	private int parent(int index) {
		if (index == 0)
			throw new IllegalArgumentException("index-0 doesn't have parent.");
		return (index - 1) / 2;
	}

	// 返回完全二叉树的数组表示中,一个索引所表示的元素的左孩子节点的索引
	private int leftChild(int index) {
		return index * 2 + 1;
	}

	// 返回完全二叉树的数组表示中,一个索引所表示的元素的右孩子结点的索引
	private int rightChild(int index) {
		return index * 2 + 2;
	}

	// 向堆中添加元素
	public void add(E e) {
		data.addLast(e);
		siftUp(data.getSize() - 1);

	}

	private void siftUp(int k) {

		while (k > 0 && data.get(parent(k)).compareTo(data.get(k)) < 0) {
			data.swap(k, parent(k));
			k = parent(k);
		}
	}

	// 看堆中的最大元素
	public E findMax() {
		if (data.getSize() == 0)
			throw new IllegalArgumentException("Can not findMax when heap is empty");
		return data.get(0);
	}

	// 取出堆中最大元素
	public E extractMax() {
		E ret = findMax();
		data.swap(0, data.getSize() - 1);
		data.removeLast();
		siftDown(0);
		return ret;
	}

	private void siftDown(int k) {
		while (leftChild(k) < data.getSize()) {
			int j = leftChild(k);
			if (j + 1 < data.getSize() && data.get(j + 1).compareTo(data.get(j)) > 0)
				j = rightChild(k);
			// data[j]是 leftChild 和 rightChild中的最大值
			if (data.get(k).compareTo(data.get(j)) >= 0)
				break;
			data.swap(k, j);
			k = j;
		}
	}

	// 取出堆中的最大元素,并且替换成元素e
	public E replace(E e) {
		E ret = findMax();
		data.set(0, e);
		siftDown(0);
		return ret;
	}

}

优先队列实现类:


public class PriorityQueue<E extends Comparable<E>> implements Queue<E> {
	private MaxHeap<E> maxHeap;
	public PriorityQueue() {
		maxHeap=new MaxHeap<>();
	}
	public int getSize() {
		return maxHeap.size();
	}
	public boolean isEmpty() {
		return maxHeap.isEmpty();
	}
	public E getFront() {
		return maxHeap.findMax();
	}
	public void enqueue(E e) {
		maxHeap.add(e);
	}
	public E dequeue() {
		return maxHeap.extractMax();
	}
}

 

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值