JAVA实现堆排序

堆排序强烈推荐这篇文章https://www.cnblogs.com/0zcl/p/6737944.html ！！

文中的堆排序用的是python实现的（循环遍历），我自己用JAVA实现了一下（循环，递归），特此记录。

为了方便阅读和使用，代码用了很多变量，看着略显繁琐，其实贼简单

核心思路：调整得到大顶堆，交换堆顶和堆底

核心数据结构：完全二叉树

核心操作：父节点和左右子节点的调整，得到大顶

核心函数：adjustTree

核心代码其实不到十行

具体代码：

二叉树类

class Tree{                 
	private int[] data;
	private int tcount;     //计数
	private int sbindex;    //sorted begin index简写，有序区的开始索引
	public Tree(int[] arrs) {
		data = arrs;
		tcount =  data.length;  
		sbindex = tcount - 1;  
		sort();
	}
	
	public void swap(int i, int j) { //交换两个节点值
		int tem = data[i];
		data[i] = data[j];
		data[j] = tem;
	}
	//根据索引调整当前节点，即父节点和左右子节点三个点的调整，递归调整得到大顶堆
	public void adjustTree(int index) { //自上而下
		int lindex = 2*(index+1)-1;
		if(lindex>sbindex) return;      //没有子节点，直接返回
		int rindex = 2*(index+1);
		int left = data[lindex];
		int parent = data[index];
		int right;       				
		if(rindex>sbindex)	            //只有左节点
			right = left - 1; 			//构造虚拟右节点，方便代码重用
		else 
			right = data[rindex];		//有右节点
		if(parent<left||parent<right) {
			if(left<right) {
				swap(index,rindex);
				adjustTree(rindex);     //递归
			}else {
				swap(index,lindex);
				adjustTree(lindex);     //递归
			}
		}
	}

	public void initTopTree() { //初始化大顶堆
		int bindex = tcount%2==0?tcount-1:tcount -2;
		for(int i=bindex;i>0;i-=2) { //自下而上
			adjustTree((i+1)/2-1);	//遍历调整非叶子节点
		}
	}
	
	public void dosort() {   //排序
		if(sbindex==0) return;
		swap(0,sbindex--);  //交换最大值和无序区最后一个值
		adjustTree(0);      //交换完之后，调整
		dosort();           //递归
	}
	
	public void sort() {
		initTopTree();  //初始化大顶堆
		dosort();       //开始排序
	}
	
	@Override
	public String toString() {  //重写toString方便打印
		return java.util.Arrays.toString(data);
	}
}

测试

public class HeapSort {
	public static void main(String[] args) {
		Tree tree = new Tree(new int[]{16,7,3,20,17,8});
		System.out.println(tree);
	}
}