堆排序(Heap Sort)
public class HeapSort { public static void sort(Comparable[] data) { // 构建最大堆 buildMaxHeap(data); // 循环,每次把根节点和最后一个节点调换位置 for (int i = data.length; i > 1; i--) { Comparable tmp = data[0]; data[0] = data[i - 1]; data[i - 1] = tmp; // 堆的长度减少1,排除置换到最后位置的根节点 maxHeapify(data, 1, i - 1); } } // 根据输入数组构建一个最大堆 private static void buildMaxHeap(Comparable[] data) { for (int i = data.length / 2; i > 0; i--) { maxHeapify(data, i, data.length); } } //堆调整,使其生成最大堆 private static void maxHeapify(Comparable[] data, int parentNodeIndex, int heapSize) { // 左子节点索引 int leftChildNodeIndex = parentNodeIndex * 2; // 右子节点索引 int rightChildNodeIndex = parentNodeIndex * 2 + 1; // 最大节点索引 int largestNodeIndex = parentNodeIndex; // 如果左子节点大于父节点,则将左子节点作为最大节点 if (leftChildNodeIndex <= heapSize && data[leftChildNodeIndex - 1].compareTo(data[parentNodeIndex - 1]) > 0) { largestNodeIndex = leftChildNodeIndex; } // 如果右子节点比最大节点还大,那么最大节点应该是右子节点 if (rightChildNodeIndex <= heapSize && data[rightChildNodeIndex - 1].compareTo(data[largestNodeIndex - 1]) > 0) { largestNodeIndex = rightChildNodeIndex; } // 最后,如果最大节点和父节点不一致,则交换他们的值 if (largestNodeIndex != parentNodeIndex) { Comparable tmp = data[parentNodeIndex - 1]; data[parentNodeIndex - 1] = data[largestNodeIndex - 1]; data[largestNodeIndex - 1] = tmp; // 交换完父节点和子节点的值,对换了值的子节点检查是否符合最大堆的特性 maxHeapify(data, largestNodeIndex, heapSize); } } }
gif 演示:http://upload.wikimedia.org/wikipedia/commons/4/4d/Heapsort-example.gif (来自wikipedia)