最大堆的实现

对于“堆”的简单描述:常见的堆为二叉堆,在二叉堆中又分为大顶堆(最顶上的元素最大)和小顶堆(最顶上的元素最小),其最终的数据结构还是一个树(完全二叉树),并且所有的节点都比它左右孩子的值大(对于大顶堆)

/**
 * 数据结构:最大堆(基于【我自定义的】链表)
 * 描述:实现了 添加、删除、判空、判断是否存在
 * @author hcc
 *
 */
public class ArrayHeap<E extends Comparable<E>> {
	private HLinkList<E> array;
	
	public ArrayHeap() {
		array = new HLinkList<E>();
	}
	
	/**
	 * 向堆中添加元素【先添加元素,在上浮到指定位置】
	 * @param e
	 */
	public void add(E e) {
		array.add(e);
		goUp();
	}
	
	/**
	 * 上浮操作
	 */
	private void goUp() {
		int index = array.getSize() - 1;
		int parentIndex = getParentIndex(index);
		while (true) {
			if (array.get(index).compareTo(array.get(parentIndex)) > 0) {
				swap(index, parentIndex);
				index = parentIndex;
				parentIndex = getParentIndex(parentIndex);
			} else {
				break;
			}
		}
	}
	
	/**
	 * 数据交换操作
	 * @param index 子节点的索引值
	 * @param parentIndex 父节点的索引值
	 */
	private void swap(int index,int parentIndex) {
		int size = array.getSize();
		if(index > (size-1) || parentIndex >  (size-1)) {
			throw new IllegalArgumentException("索引值错误!");
		}
		E temporary = array.get(index);
		array.set(index, array.get(parentIndex));
		array.set(parentIndex, temporary);
		
	}
	
	/**
	 * 通过index节点的索引获取父节点的索引
	 * @param index 
	 * @return 父节点的索引
	 */
	private int getParentIndex(int index) {
		if(index < 0) {
			throw new IllegalArgumentException("索引的值小于0!");
		}
		return (index-1)/2;
	}
	
	/**
	 * 通过index节点的索引获取左子节点的索引
	 * @param index
	 * @return
	 */
	private int getLeftIndex(int index) {
		if(index < 0) {
			throw new IllegalArgumentException("索引的值小于0!");
		}
		return (2*index+1);
	}
	
	/**
	 * 通过index节点的索引获取有子节点的索引
	 * @param index
	 * @return
	 */
	private int getRightIndex(int index) {
		if(index < 0) {
			throw new IllegalArgumentException("索引的值小于0!");
		}
		return (2*index+2);
	}
	
	/**
	 * 
	 * @return 返回堆中最大的数据
	 */
	public E findMax() {
		if(array.getSize() == 0) {
			throw new IllegalArgumentException("堆中无数据!!!");
		}
		return array.get(0);
	}
	/**
	 * 从堆中取出最大数【取出后堆的排列顺序发生变化,需要在删除后的堆中再找到最大数】
	 */
	public E getMax() {
		E e = findMax();
		swap(0,array.getSize()-1);
		array.remove(array.getSize()-1);
		moveDown(0);
		return e;
	}
	
	/**
	 * 下移操作
	 * @param index
	 */
	private void moveDown(int index) {
		if(index < 0) {
			 throw new IllegalArgumentException("索引的值小于0!");
		}
		while(getLeftIndex(index) < array.getSize()) {
			int i = 0;
			E e = array.get(index);
			int childLeft = getLeftIndex(index);
			int childRight = getRightIndex(index);
			
			E childLeftData = array.get(childLeft);
			
			i = childLeft;
			if(childRight < array.getSize() && array.get(childRight).compareTo(childLeftData) >= 0) {
				i = childRight;
			}
			if(e.compareTo(array.get(i)) >= 0) {
				break;
			}
			swap(index,i);
			index = i;
		}	
	}
	/**
	 * 
	 * @return 数据数量
	 */
	public int getSize() {
		return array.getSize();
	}
	/**
	 * 
	 * @param e
	 * @return true表示堆中存在 false表示堆中不存在
	 */
	public boolean contain(E e) {
		return array.contains(e);
	}
	/**
	 * 
	 * @return true表示为空 false表示不为空
	 */
	public boolean isEmpty() {
		return array.isEmpty();
	}
	
	
	public String toString() {
		return array.toString();
	}
}

 

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值