图解最小堆形成(附带java实现代码)

最小(大)堆

最小(大)堆是一颗完全二叉树,该树中的某个节点的值总是不大于(不小于)其左右子节点的值。可以通过下图理解,另外,为什么会使用数组来保存呢?因为利用完全二叉树的性质,我们可以通过数组来表示完全二叉树(数组下标与完全二叉树节点存在映射关系,比如父节点可以通过Math.floor((index-1)/2)来获取),从而简化了实现及开销,避免使用额外的指针来实现树结构。

最小(大)堆性质

  • 树根节点的值是所有堆节点值中最小(大)值。
  • 树中每个节点的子树也都是最小(大)堆。

最小(大)堆作用

  • 最小(大)堆能保证堆顶元素为最小,而如果使用数组无法达到该效果。数组如果要访问最小值则需要遍历查找最小值,时间复杂度至少O(n)。而最小堆访问最小值时间复杂度为O(1),当然天底下没有免费的午餐,我们需要做额外的工作去维护最小(大)堆的结构,这也是需要复杂度花销的。

当然这也是最小(大)堆的优势,通过动态维护使得最小值的获取代价很小,实际上维护的时间复杂度为O(logN)。而数组则无法做到如此,如果数组想要维护顺序性则需要的复杂度至少为O(N)。这样来看最小(大)堆的优势就凸现出来了。

插入操作

为避免冗长累赘,我们这里只挑最小堆作为例子进行说明,最大堆的情况与最大堆相似。

现在分别插入4 7 2 5 6 1 0 3 8,使用一个数组来保存最小堆,为了帮助理解,数组下方提供一个逻辑上的完全二叉树的结构,两者结合着更容易理解其中机制。首先插入4,

接着插入7,插入后检测到树符合最小堆要求,所以不改动。

继续插入2,插入后检测到不符合最小堆要求,父节点4大于右子节点2,

于是将它们对调。

继续插入5,插入后检测到不符合最小堆要求,父节点7大于左子节点5,

于是将它们对调。

继续插入6,插入后检测到树符合最小堆要求,所以不改动。

继续插入1,插入后检测到不符合最小堆要求,父节点4大于左子节点1,

于是将它们对调,

对调后继续检测到不符合最小堆要求,父节点2大于右子节点1,

继续将它们对调。

继续插入0,插入后检测到不符合最小堆要求,父节点2大于右子节点0,

于是将它们对调,

对调后继续检测到不符合最小堆要求,父节点1大于右子节点0,

继续将它们对调。

继续插入3,插入后检测到不符合最小堆要求,父节点7大于左子节点3,

于是将它们对调,

对调后继续检测到不符合最小堆要求,父节点5大于左子节点3,

继续将它们对调,然后符合最小堆要求,不必继续往上对调。

继续插入8,插入后检测到树符合最小堆要求,所以不改动。以上,完成所有元素的最小堆插入操作。

删除操作

删除操作其实就是删除最小值,即最小堆树中的根节点。主要是将树中最后一个节点替换到被删除的根节点,然后自顶向下递归调整使之符合最小堆要求。

删除根节点0,然后将树的最后一个节点8补到根节点上。

比较根节点的左右子节点,

因为右子节点1比较小,所以我们要进一步比较的是根节点8与右子节点1,

1小于8,于是对调。

继续比较现在节点8的左右子节点,

因为右子节点2比较小,所以我们要进一步比较的是根节点8与右子节点2,

2小于8,于是对调。

至此,完成最小值删除操作。

完成的代码实例;

import java.util.Comparator;

public class MinHeap<T> {
	T[] arr;
	private static int capacity = Integer.MAX_VALUE;
	private Comparator<T> comparator;

	private int currentIndex = 0;

	public MinHeap(int capacity, Comparator<T> comparator) {
		if (capacity < 1) {
			throw new RuntimeException("error");
		}

		arr = (T[]) new Object[capacity];
		MinHeap.capacity = capacity;
		this.comparator = comparator;
	}

	public MinHeap(Comparator<T> comparator) {
		this(capacity, comparator);
	}

	public void add(T t) {
		if (currentIndex >= capacity) {
			// 超出容量更新
			update(t);
			return;
		}
		// 插入节点
		doAdd(t);
	}

	private void update(T add) {
		removeMin();
		doAdd(add);
	}

	private void doAdd(T t) {
		// 第一个节点
		if (currentIndex < 1) {
			arr[0] = t;
			currentIndex++;
			return;
		}
		// 插入到尾部,然后平衡
		arr[currentIndex] = t;
		addBalance(currentIndex);
		currentIndex++;
	}

	private void addBalance(int index) {
		// 获取父节点
		int parentIndex = getParentIndex(index);
		if (parentIndex < 0) {
			return;
		}
		// 比较父节点和和自己,小于就替换
		if (comparator.compare(arr[parentIndex], arr[index]) > 0) {
			replace(parentIndex, index);
		}
		// 再次炒父节点平衡
		addBalance(parentIndex);

	}

	public static void main(String[] args) {
		MinHeap<Integer> m = new MinHeap<Integer>(9, (o1, o2) -> {
			return o1 - o2;
		});
		m.add(4);
		m.add(7);
		m.add(2);
		m.add(5);
		m.add(6);
		m.add(1);
		m.add(0);
		m.add(3);
		m.add(8);
		m.removeMin();
	}

	private void replace(int index, int index1) {
		T t = arr[index];
		arr[index] = arr[index1];
		arr[index1] = t;
	}

	/**
	 * 删除首节点 把尾节点放到首节点,然后平衡
	 * 
	 * @return
	 */
	public T removeMin() {
		if (currentIndex < 1) {
			throw new RuntimeException();
		}
		T t = arr[0];
		// 把尾节点放到首节点
		arr[0] = arr[currentIndex - 1];
		// 平衡
		removeMinBalance(0);
		currentIndex--;
		System.err.println(Arrays.toString(arr));
		return t;
	}

	private void removeMinBalance(int index) {
		// 获取子节点
		int childIndex = getChildIndex(index);
		if (childIndex >= currentIndex) {
			return;
		}
		int minIndex = 0;
		// 找出最小的子节点
		if (childIndex + 1 < currentIndex) {
			if (comparator.compare(arr[childIndex], arr[childIndex + 1]) > 0) {
				minIndex = childIndex + 1;
			}
		} else {
			minIndex = childIndex;
		}
		// 最小的子节点和父节点比较,大于子节点就替换
		if (comparator.compare(arr[index], arr[minIndex]) > 0) {
			replace(minIndex, index);
		} else {
			return;
		}
		// 再次平衡
		removeMinBalance(minIndex);
	}

	private int getParentIndex(int index) {
		if (index == 0) {
			return -1;
		}
		return (index - 1) / 2;
	}

	private int getChildIndex(int index) {
		return index * 2 + 1;
	}

}
  • 1
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值