代码
public class Main {
public static void main(String[] args) {
int[] arr = {9,8,7,6,5,4,3,2,1};
sort(arr);
System.out.println(Arrays.toString(arr));
}
public static void sort(int[] arr){
//完全二叉树 以数组方式存放 则第一个非叶子节点下标为 arr.length/2 - 1
//首先创建大顶堆
for (int i = arr.length / 2 - 1;i >= 0;i--){
adjustHeap(arr,i,arr.length);
}
//然后进行堆排序
for (int j = arr.length - 1;j > 0;j--){
int temp = arr[j];//把最后一个元素保存起来
arr[j] = arr[0];
arr[0] = temp;
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 j = 2 * i + 1;j < length;j = j * 2 + 1){
//先判断左右子节点的大小
if (j + 1 < length && arr[j] < arr[j + 1]){
j++;//j指向大的值
}
//再判断j指向的子节点和i指向的父节点的大小
if (temp < arr[j]){//注意 这里是temp与arr[j] 进行比较 而不是arr[i] 和 arr[j]
//需要给父节点赋值
arr[i] = arr[j];
i = j;//将i指向刚更换完的节点
}else {
break;
}
}
//将i指向的最后节点赋值
arr[i] = temp;
}
}
结果
[1, 2, 3, 4, 5, 6, 7, 8, 9]
完整版【包括小堆排序】
public class Main {
public static void main(String[] args) {
int[] arr = {9,8,7,6,5,4,3,2,1};
sort(arr);
int[] arr2 = {1,4,3,5,2,8,7,9,6};
sortSmall(arr2);
System.out.println(Arrays.toString(arr2));
}
//小堆排序
public static void sortSmall(int[] arr){
//构建小顶堆
for (int j = arr.length / 2 - 1;j >= 0;j--){
adjustSmallHeap(arr,j,arr.length);
}
//进行排序
for (int j = arr.length - 1;j > 0;j--){
int temp = arr[0];
arr[0] = arr[j];
arr[j] = temp;
//System.out.println("hahah1"+Arrays.toString(arr));
adjustSmallHeap(arr,0,j);
//System.out.println("hahah2"+Arrays.toString(arr));
}
}
//构建小顶堆
public static void adjustSmallHeap(int[] arr,int i,int length){
int temp = arr[i];
for (int k = i*2 + 1;k < length;k = k*2 + 1){
//先让k定位到小的
if (k + 1 < length && arr[k + 1] < arr[k]){
k++;
}
//然后考虑赋值
if (temp > arr[k]){
arr[i] = arr[k];
i = k;
}else {
break;
}
}
arr[i] = temp;//!!!! 注意 这个千万不能忘
}
//大堆排序
public static void sort(int[] arr){
//完全二叉树 以数组方式存放 则第一个非叶子节点下标为 arr.length/2 - 1
//首先创建大顶堆
for (int i = arr.length / 2 - 1;i >= 0;i--){
adjustHeap(arr,i,arr.length);
}
//然后进行堆排序
for (int j = arr.length - 1;j > 0;j--){
int temp = arr[j];//把最后一个元素保存起来
arr[j] = arr[0];
arr[0] = temp;
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 j = 2 * i + 1;j < length;j = j * 2 + 1){
//先判断左右子节点的大小
if (j + 1 < length && arr[j] < arr[j + 1]){
j++;//j指向大的值
}
//再判断j指向的子节点和i指向的父节点的大小
if (temp < arr[j]){//注意 这里是temp与arr[j] 进行比较 而不是arr[i] 和 arr[j]
//需要给父节点赋值
arr[i] = arr[j];
i = j;//将i指向刚更换完的节点
}else {
break;
}
}
//将i指向的最后节点赋值
arr[i] = temp;
}
}
结果
[9, 8, 7, 6, 5, 4, 3, 2, 1]
一定要注意arr[i] = temp;