i 结点:
其左结点位置:2 * i + 1
其右结点的位置:2 * i + 2
其父结点的位置:(i - 1) / 2
/*
1)将整个数组调整为一个大顶堆(此时第一个元素是最大的)
2)然后将第一个元素与最后一个元素交换位置(然后删除最后一个元素。通过减小length的方式)
3)将剩余元素再次构造成一个大顶堆,然后反复执行,最后即可得到一个有序数组
*/
public class DuiPai {
public static void main(String[] args) {
int[] a = {4, 5, 8, 2, 3, 9, 7, 1};
//数组升序排列 大顶堆
heapSort(a);
for (int i = 0; i < a.length; i++) {
System.out.print(a[i] + " ");
}
}
public static void heapSort(int[] a) {
//将整个数组构造成大顶堆(从第一个非叶子结点开始调整)
for (int i = a.length / 2 - 1; i >= 0; i--) {
adjustHeap(a, i, a.length);
}
for (int j = a.length - 1; j > 0; j--) {
//将第一个元素(最大的元素)与a[j]互换位置,即和最后一个元素互换位置(注意:此处最后一个元素不是指数组的最后一个元素a[7])
int t = a[j];
a[j] = a[0];
a[0] = t;
//重新构造大顶堆,因为此时第一个元素已经不是最大的了,需要重新将数组构造成大顶堆
adjustHeap(a, 0, j);
}
}
//将一个数组,调整成一个大顶堆
/**
* i 表示非叶子结点在数组中的索引
* length 表示对多少个元素进行调整 length在逐渐减小
*/
public static void adjustHeap(int[] a, int i, int length) {
int temp = a[i];
for (int k = 2 * i + 1; k < length; k = 2 * k + 1) {
if (k + 1 < length && a[k] < a[k + 1]) {
k++;
}
if (a[k] > temp) {
a[i] = a[k];
i = k;
} else {
break;
}
}
a[i] = temp;
}
}