/**
* 升序用大根堆,降序用小根堆
* 调整堆:从根节点开始向下调整堆
* ①先建立初始堆,从n/2-1(第一个非叶子节点)节点开始,将之看做根节点,向下调整堆,到根节点结束
* ②循环i from n-1 to 0,每次取出堆的最顶层根节点,即数组下标为0,然后与节点i交换,这样对于大根堆而言,最大值总是在后面,再从根节点开始调整堆
*/
public class HeapSort {
public static void main(String[] args) {
int[] a = { 26, 5, 77, 1, 61, 11, 59, 15, 48, 19 };
Sort(a);
}
//heapSort
public static void Sort(int[] a) {
int n = a.length;//取得数组总长度,及堆最大的序号
int temp = 0;
Display(a, "Before sort : ");
for (int i = n / 2 - 1; i >= 0; i--) {
Adjust(a, i, n);
Display(a, "create MaxHeap : ");
}
System.out.println("---------------------------------------");
for (int i = n - 2; i >= 0; i--) {
temp = a[i + 1];//取出最后一个元素
a[i + 1] = a[0];//取出第一个元素,即顶层根节点
a[0] = temp;//交换位置
Adjust(a, 0, i + 1);//调整堆
Display(a, "reCreate : ");
}
Display(a, "After sort : ");
}
/**
* 调整堆的方法
* @param a 要调整的数组,即堆
* @param i 调整的根节点,即起始位置
* @param n 要调整的终止位置
*/
public static void Adjust(int[] a, int i, int n) {
int j = 0;
int temp = 0;
temp = a[i]; //取出根节点
j = 2 * i + 1; //左孩子节点
while (j <= n - 1) {
if (j < n - 1 && a[j] < a[j + 1])//比较左右孩子,取出较大的孩子
j++;
if (temp >= a[j]) //如果根节点大于孩子节点则退出循环,不用调整
break;
a[(j - 1) / 2] = a[j];//较大的孩子节点值赋值给根节点
j = 2 * j + 1;//继续寻找左孩子
}
a[(j - 1) / 2] = temp;//将根节点赋值给最后一个空出来的节点
}
//打印堆内容
public static void Display(int[] a, String str) {
System.out.println(str);
for (int i = 0; i < a.length; i++)
System.out.print(a[i] + " ");
System.out.println();
}
}
堆排序java实现
最新推荐文章于 2024-07-27 17:56:15 发布