public static void main(String[] args) {
//数组我们从下标0开始计算,那么就满足规则:节点i的左子结点:2i+1,右子节点:2i+2
//节点i的父节点为:(i-1)/2
int[] a = {3,2,5,4,6,7,8,9,2,0,5,3,1,6,7,8,4,3};
heapSort(a);
for (int i = 0; i < a.length; i++) {
System.out.println(a[i]);
}
}
/**
* 堆排序
* @param a 源数组
*/
public static void heapSort(int[] a) {
//a.length-1为最后一个节点的编号,再减去1然后除以2就得到最后一个节点的父节点
int n = (a.length-1-1) / 2;
//构建初始堆
for (int i = n; i >= 0; i--) {
rebuild(a, i, a.length - 1);
}
for (int i = a.length - 1; i >= 1; i--) {
int tmp = a[0];
a[0] = a[i];
a[i] = tmp;
rebuild(a, 0, i - 1);
}
}
/**
* 重建堆
* @param a 源数组
* @param k 当前节点
* @param end 最大可用节点编号
*/
public static void rebuild(int[] a, int k, int end) {
int i = k, j = 2 * k + 1;
int temp = a[k];
while (j <= end) {
if (j < end && a[j] < a[j+1]) { // 如果j=a.length-1的话就没有右子节点了
j++;
}
if (temp < a[j]) {
a[i] = a[j];
i = j;
j = 2 * j + 1;
} else {
break;
}
}
a[i] = temp;
}
java 堆排序
最新推荐文章于 2023-02-27 18:46:36 发布