java实现
大跟堆
小跟堆实现的堆排序,获取最小的K个数的值。
输出:0 1 2 3 4
大跟堆
package com.practice;
public class BigHeap {
public void shift(int[] arr, int k, int m) {
int i = k, j = 2 * i + 1;
while(j<m) {
if(j<m-1 && arr[j]<arr[j+1]) j++;
if(arr[i] > arr[j]) break;
else {
swap(arr, i, j);
i = j;
j = 2*i + 1;
}
}
}
private void swap(int[] arr, int i, int j) {
int temp = arr[i];
arr[i] = arr[j];
arr[j] = temp;
}
public void sort(int[] arr, int n) {
for(int i=n/2 -1;i>=0;i--)
shift(arr, i, n);
for(int i=0;i<n;i++) {
System.out.print(arr[0]+",");
swap(arr, 0, n-i-1);
shift(arr, 0, n-i-1);
}
}
}
小跟堆实现的堆排序,获取最小的K个数的值。
package com.practice;
public class HeapSort {
public void sift(int arr[], int k, int m) {
int i = k, j = 2 * i + 1;
while(j<m) {
if(j<m-1 && arr[j]>arr[j+1]) j++;
if(arr[i]<arr[j]) break;
else {
swap(arr, i, j);
i = j;
j = 2 * i +1;
}
}
}
private void swap(int[] arr, int i, int j) {
int temp = arr[i];
arr[i] = arr[j];
arr[j] = temp;
}
public void sort(int[] arr, int n) {
for(int i=n/2-1;i>=0;i--) {
sift(arr, i, n);
}
}
public void getTopK(int[]arr, int k) {
sort(arr, arr.length);
for(int i=0;i<k;i++) {
System.out.print(arr[0]+" ");
swap(arr, 0, arr.length-i-1);
sift(arr, 0, arr.length-i-1);
}
}
public static void main(String[] args) {
int[] arr = { 2, 7, 8, 3, 1, 6, 9, 0, 5, 4 };
new HeapSort().getTopK(arr, 5);;
}
}
输出:0 1 2 3 4