java简易优先队列（小顶堆）

原创于 2021-03-21 13:23:39 发布 · 340 阅读

0 ·

CC 4.0 BY-SA版权

基础算法与数据结构专栏收录该内容

3 篇文章

订阅专栏

本文介绍了一个基于Java实现的小顶堆优先队列。该队列通过自定义的Heap类实现，支持插入元素（offer）及弹出最小元素（poll）。文章详细展示了如何维护堆的性质，并提供了完整的源代码。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

java简易优先队列（小顶堆）

class Heap{
	private int[] heap;
	private int size;
	public Heap(int n)
	{
		heap = new int[n + 1];
		size = 0;
	}

	public void offer(int num)
	{
		int father = ((size - 1) >> 1), index = size;
		heap[index] = num;
		while (father >= 0)
		{
			if (heap[father] > heap[index])	swap(index, father);
			index = father;
			father = ((father - 1) >> 1);
		}
		size++;
	}

	public int poll()
	{
		int res = heap[0];
		swap(0, size - 1);
		int temp = heap[0];
		for (int i = 1;i < size - 1;i = (i << 1) + 1)
		{
			if (i + 1 < size - 1 && heap[i] > heap[i + 1])  i++;
			if (temp < heap[i]) break;
			heap[((i - 1) >> 1)] = heap[i];
			heap[i] = temp;
		}
		size--;
		return res;
	}

	private void swap(int indexA, int indexB)
	{
		int temp = heap[indexA];
		heap[indexA] = heap[indexB];
		heap[indexB] = temp;
	}
}