如何在Java中的数组上执行堆排序?
以下是堆排序(maxheap)的算法。
步骤1-在堆的末尾创建一个新节点。
步骤2-为节点分配新值。
步骤3-比较此子节点与其父节点的值。
步骤4-如果parent的值小于孩子,则交换它们。
步骤5-重复步骤3和4,直到保持“堆”属性。
示例
import java.util.Arrays; import java.util.Scanner; public class Heapsort { public static void heapSort(int[] myArray, int length) { int temp; int size = length-1; for (int i = (length / 2); i >= 0; i--) { heapify(myArray, i, size); }; for(int i= size; i>=0; i--) { temp = myArray[0]; myArray[0] = myArray[size]; myArray[size] = temp; size--; heapify(myArray, 0, size); } System.out.println(Arrays.toString(myArray)); } public static void heapify (int [] myArray, int i, int heapSize) { int a = 2*i; int b = 2*i+1; int largestElement; if (a<= heapSize && myArray[a] > myArray[i]) { largestElement = a; } else { largestElement = i; } if (b <= heapSize && myArray[b] > myArray[largestElement]) { largestElement = b; } if (largestElement != i) { int temp = myArray[i]; myArray[i] = myArray[largestElement]; myArray[largestElement] = temp; heapify(myArray, largestElement, heapSize); } } public static void main(String args[]) { Scanner scanner = new Scanner(System.in); System.out.println("Enter the size of the array :: "); int size = scanner.nextInt(); System.out.println("Enter the elements of the array :: "); int[] myArray = new int[size]; for(int i=0; i<size; i++) { myArray[i] = scanner.nextInt(); } heapSort(myArray, size); } }
输出结果
Enter the size of the array :: 5 Enter the elements of the array :: 45 125 44 78 1 [1, 44, 45, 78, 125]