数据结构与算法之堆排序

1、堆排序简介

        堆排序（英语：Heapsort）是指利用堆这种数据结构所设计的一种排序算法。堆是一个近似完全二叉树的结构，并同时满足堆的性质：即子结点的键值或索引总是小于（或者大于）它的父节点。

2、堆排序思想

        假设一个堆的结构如下：

        （1）调整：

        从最后个非叶子节点[arr.length / 2 -1]开始也就是索引为1的这个节点开始调整以它为根节点的树。且以一个临时变量保存这个节点的数据，即：int temp = arr[arr.length / 2 -1]；继续判断下面如果无子节点，则arr不做任何变化。如果有子节点且继续判断子节点的和temp的大小关系，最后将最大的数与temp交换。

 

         通过循环索引自减来到索引为0的这个节点，将最大的数据调整到堆顶。判断下方的子节点有无字节点，使其形成一个大顶堆（每个父节点都大于它的子节点）。

 

/**
	 * 功能：完成将i指定的非叶子节点调整为大顶堆
	 * @param arr 待调整的数组
	 * @param i 表示非叶子节点在数组中的索引
	 * @param length // 表示需要调整的元素个数
	 */
	public static void adjustHeap(int arr[], int i, int length) {
		
		// 取出当前元素的值
		int temp = arr[i];
		
		for(int k = i * 2 + 1; k < length; k = k * 2 + 1) {
			if(k+1 < length && arr[k] < arr[k+1]) { // 说明右子节点大于左子节点
				k++; // 使得k 执行右子节点
			}
			if(temp < arr[k]) { // 如果子节点大于父节点
				arr[i] = arr[k];
				i = k;
			}else {
				break;
			}
		}
		arr[i] = temp;
		
	}

 

         （2）交换

        此时arr由{3，4，5，1，2}变为了{5，4，3，1，2}，我们将arr[0]与arr[lengh - 1]更换位置变为{2, 4, 3, 1, 5}，更换之后不再考虑索引为（lengh - 1）的节点。

 

         之后继续进行调整和交换直到有序。

         此时变成了一个有序的序列。{1， 2，3，4，5}

public static void heapSort(int[] arr) {
		System.out.println("堆排序！");
		
		int j = 0, h = 0;
		
		for(int i = arr.length / 2 - 1; i >= 0; i-- ) {
			adjustHeap(arr, i, arr.length);
			System.out.println("第"+(++j)+"次运行" + Arrays.toString(arr));
		}
		
		System.out.println();
		
		// 进行交换
		for(int q = arr.length - 1; q > 0; q--) {
			
			int temp = arr[q];
			arr[q] = arr[0];
			arr[0] = temp;
			System.out.println("第"+(++h)+"次运行" + Arrays.toString(arr));
			adjustHeap(arr, 0, q);
			
		}
		
	}

3、测试

public static void main(String[] args) {
		// 要求将数组进行升序排列
		int arr[] = {4, 6, 8, 5, 9, -2, 99, 89, 103};
		heapSort(arr);
	}

 4、完整代码

package tree;

import java.util.Arrays;

public class HeapSort2 {

	public static void main(String[] args) {
		// 要求将数组进行升序排列
		int arr[] = { 4, 6, 8, 5, 9, -2, 99, 89, 103 };
		heapSort2(arr);
		System.out.println(Arrays.toString(arr));
	}

	public static void heapSort2(int[] arr) {

		for (int i = arr.length / 2 - 1; i >= 0; i--) {
			toBigTopHeap(arr, i, arr.length);
		}

		// 交换&调整
		for (int j = arr.length - 1; j > 0; j--) {
			int temp = arr[j];
			arr[j] = arr[0];
			arr[0] = temp;
			toBigTopHeap(arr, 0, j);
		}

	}

	// 定义方法使得树变为大顶堆
	public static void toBigTopHeap(int[] arr, int i, int length) {

		int temp = arr[i];

		for (int k = i * 2 + 1; k < length; k = k * 2 + 1) {
			if (k + 1 < length && arr[k] < arr[k + 1]) {
				k++;
			}
			if (arr[k] > temp) { // 如果当前元素大于非叶子节点
				arr[i] = arr[k];
				i = k;
			} else {
				break;
			}
		}
		arr[i] = temp;

	}

}