堆排序(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) 

posted @ 2017-03-13 22:21  Allen Wen  阅读(228)  评论(0编辑  收藏  举报