Java数据结构_10.常用排序算法_堆排序

堆排序

原理：将数组转为完全二叉树再转为大顶堆，然后将根节点与叶子节点的最后一个值调换，找到最大值，将最大值取出，现在又变成一个完全二叉树，继续将这个完全二叉树转成大顶堆，以此类推，直至找到所有的最大值。

升序排序用大顶堆，降序排序用小顶堆

大顶堆:父节点总比两个子节点的权值要大

package com.paixu;

import java.util.Arrays;

public class HeapSort {

	public static void main(String[] args) {
		int arr[] = new int[] { 9, 6, 8, 7, 1, 0, 10, 4, 2 };
		heapSort(arr);
		System.out.println(Arrays.toString(arr));// [0, 1, 2, 4, 6, 7, 8, 9, 10]
	}

	public static void heapSort(int[] arr) {
		// 开始位置是最后一个非叶子节点，即最后一个节点的父节点
		int start = (arr.length - 1) / 2;
		// 调整为大顶堆
		for (int i = start; i >= 0; i--) {
			maxHeap(arr, arr.length, i);
		}
		// 把数组中的第0个和堆中的最后一个交换位置，再把前面的处理为大顶堆
		for (int i = arr.length - 1; i > 0; i--) {
			int temp = arr[0];
			arr[0] = arr[i];
			arr[i] = temp;
			maxHeap(arr, i, 0);
		}
	}

	public static void maxHeap(int[] arr, int size, int index) {
		// 左子节点
		int leftNode = 2 * index + 1;
		// 右子节点
		int rightNode = 2 * index + 2;
		int max = index;
		// 和两个子节点分别对比，找出最大的节点
		if (leftNode < size && arr[leftNode] > arr[max]) {
			max = leftNode;
		}
		if (rightNode < size && arr[rightNode] > arr[max]) {
			max = rightNode;
		}
		// 交换位置
		if (max != index) {
			int temp = arr[index];
			arr[index] = arr[max];
			arr[max] = temp;
			// 交换位置后，可能会破坏之前排好的堆，所以之前排好的堆要重新调整
			maxHeap(arr, size, max);
		}
	}
}