堆的定义:
堆实际上是一棵完全二叉树。
堆满足两个性质:
- 堆的每一个父节点都大于(或小于)其子节点;
- 堆的每个左子树和右子树也是一个堆。 堆的每个左子树和右子树也是一个堆。
堆的分类:
堆分为两类:
- 最大堆(大顶堆):堆的每个父节点都大于其孩子节点;
- 最小堆(小顶堆):堆的每个父节点都小于其孩子节点;
公式来描述一下堆的定义:
大顶堆:arr[i] >= arr[2i+1] && arr[i] >= arr[2i+2]
小顶堆:arr[i] <= arr[2i+1] && arr[i] <= arr[2i+2]
堆排序的基本思想:
将待排序序列构造成一个大顶堆,此时,整个序列的最大值就是堆顶的根节点。将其与末尾元素进行交换,此时末尾就为最大值。然后将剩余n-1个元素重新构造成一个堆,这样会得到n个元素的次小值。如此反复执行,便能得到一个有序序列了
代码演示:
package com.study.heap;
import java.util.Arrays;
/**
* @Title: HeapSort
* @Description: 堆排序
* @author: youqing
* @version: 1.0
* @date: 2018/11/16 16:57
*/
public class HeapSort {
public static void main(String []args){
int []arr = {9,5,4,1,3,7,8,2,6};
sort(arr);
System.out.println(Arrays.toString(arr));
}
public static void sort(int []arr){
//1.构建大顶堆
for(int i = arr.length/2-1;i>=0;i--){
//从第一个非叶子结点从下至上,从右至左调整结构
adjustHeap(arr,i,arr.length);
}
//2.调整堆结构+交换堆顶元素与末尾元素
for (int j = arr.length - 1;j>0;j--){
swap(arr,0,j);//将堆顶元素与末尾元素进行交换
adjustHeap(arr,0,j);//重新对堆进行调整
}
}
/**
* 调整大顶堆(仅是调整过程,建立在大顶堆已构建的基础上)
* @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++;
}
if(arr[k]>temp){
arr[i] = arr[k];
i = k;
}else{
break;
}
}
arr[i] = temp;
}
/**
* 交换元素
* @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;
}
}