堆排序,就是利用堆快速的选出最小值(小堆顶)或最大值(大堆顶),并保证所有的节点值都小于或大于其子节点。每次选出的堆顶的值,肯定是最小或最大的值,放在数组尾部(下次不参与堆),实际上来说就是通过小堆顶或者大堆顶加快选出最小最大值。
具体过程建立小堆顶或大堆顶函数,函数从最后一个非叶子节点开始,若子节点比当前节点的值小或者大则交换,并在交换以后,将当前坐标的交换到的位置作为节点选出其子节点中的最小或者最大值替换当前的值()(后续的步骤是非必要的,目的是为了尽量让大值或者小值沉下去)。直到堆顶就完成了一轮小堆顶,大堆顶的建立,并得到了最小值或者最大值。然后将最小值或最大值与数字最后一位交换。这样就固化了数字尾部的值,下一次数组尾部的值就不参与堆排序了。
/**
* 选择排序-堆排序
* @param array 待排序数组
* @return 已排序数组
*/
public static int[] baiduHeapSort(int[] array) {
//这里元素的索引是从0开始的,所以最后一个非叶子结点array.length/2 - 1
for (int i = array.length / 2 - 1; i >= 0; i--) {
baiduAdjustHeap(array, i, array.length); //调整堆
}
// 上述逻辑,建堆结束
// 下面,开始排序逻辑
for (int j = array.length - 1; j > 0; j--) {
System.out.print(j+":排序之后==>");
for(int i=0;i<array.length;i++){
System.out.print(array[i]+" ");
}
System.out.println();
// 元素交换,作用是去掉大顶堆
// 把大顶堆的根元素,放到数组的最后;换句话说,就是每一次的堆调整之后,都会有一个元素到达自己的最终位置
swap(array, 0, j);
// 元素交换之后,毫无疑问,最后一个元素无需再考虑排序问题了。
// 接下来我们需要排序的,就是已经去掉了部分元素的堆了,这也是为什么此方法放在循环里的原因
// 而这里,实质上是自上而下,自左向右进行调整的
baiduAdjustHeap(array, 0, j);
}
return array;
}
/**
* 优点是每次都保证了父节点大于子节点,写法不好,交换以后,取的一般都很小,然后就会导致下一波移动次数过于多,
* 可以增加偏移量,不把大数移到后面去,同时也使用小根堆
* 整个堆排序最关键的地方
* @param array 待组堆
* @param i 起始结点
* @param length 堆的长度
*/
public static void baiduAdjustHeap(int[] array, int i, int length) {
// 先把当前元素取出来,因为当前元素可能要一直移动
int temp = array[i];
for (int k = 2 * i + 1; k < length; k = 2 * k + 1) { //2*i+1为左子树i的左子树(因为i是从0开始的),2*k+1为k的左子树
// 让k先指向子节点中最大的节点
if (k + 1 < length && array[k] < array[k + 1]) { //如果有右子树,并且右子树大于左子树
k++;
}
//如果发现结点(左右子结点)大于根结点,则进行值的交换
if (array[k] > temp) {
swap(array, i, k);
// 如果子节点更换了,那么,以子节点为根的子树会受到影响,所以,循环对子节点所在的树继续进行判断
i = k;
} else { //不用交换,直接终止循环
break;
}
}
}
/**
* 交换元素
* @param arr
* @param a 元素的下标
* @param b 元素的下标
*/
public static void swap(int[] arr, int a, int b) {
int temp = arr[a];
arr[a] = arr[b];
arr[b] = temp;
}