堆排序

作者: wayneinyz | 来源:发表于2017-11-02 12:12 被阅读6次
    public class HeapSort {
    
        /*
        堆排序是一种树形选择排序,是对直接选择排序的有效改进。
         */
    
        public static void heapSort(int[] a) {
            int arrayLength = a.length;
            // 循环建堆
            for (int i = 0; i < arrayLength - 1; i++) {
                // 建堆
                buildMaxHeap(a, arrayLength - 1 - i);
                // 交换堆顶和最后一个元素
                swap(a, 0, arrayLength - 1 - i);
            }
        }
    
        // 对data数组从0到lastIndex建大顶堆
        public static void buildMaxHeap(int[] data, int lastIndex) {
            // 从lastIndex处节点(最后一个节点)的父节点开始
            for (int i = (lastIndex - 1) / 2; i >= 0; i--) {
                // k保存正在判断的节点
                int k = i;
                // 如果当前k节点的子节点存在
                while (k*2 + 1 <= lastIndex) {
                    // k节点的左子节点的索引
                    int biggerIndex = 2*k + 1;
                    // 如果biggerIndex小于lastIndex,即biggerIndex+1代表的k节点的右子节点存在
                    if (biggerIndex < lastIndex) {
                        // 若果右子节点的值较大
                        if (data[biggerIndex] < data[biggerIndex + 1])
                            // biggerIndex总是记录较大子节点的索引
                            biggerIndex++;
                    }
                    // 如果k节点的值小于其较大的子节点的值
                    if (data[k] < data[biggerIndex]) {
                        // 交换他们
                        swap(data, k, biggerIndex);
                        // 将biggerIndex赋予k,开始while循环的下一次循环,
                        // 重新保证k节点的值大于其左右子节点的值
                        k = biggerIndex;
                    } else
                        break;
                }
            }
        }
    
        public static void swap(int[] a, int i, int j) {
            int temp = a[i];
            a[i] = a[j];
            a[j] = temp;
        }
    
        public static void main(String[] args) {
            int[] a = new int[] {2, 4, 7, 5, 11, 3, 1, 9, 7, 8, 10, 6, -1, 0, -2, 12};
            heapSort(a);
            for (int i = 0; i < a.length; i++) {
                System.out.print(a[i] + " ");
            }
        }
    
    }
    

    相关文章

      网友评论

          本文标题:堆排序

          本文链接:https://www.haomeiwen.com/subject/zwyopxtx.html