Kotlin练习-堆排序

Java版本

object HeapSort {    /**     * 创建最大堆结构,要求是任何非叶子节点都比它的左右两个节点都要大     */    fun createHeap(array: MutableList<Int>, i: Int) {        if (array.size == 1) return        //首先堆的重要特性是非叶子节点数少于size/2，所以我们只需要调整前size/2索引上的值        val leftChildIndex = 2 * i + 1        val rightChildIndex = 2 * i + 2        var max = i        if (i <= array.size / 2) {            //如果左子节点比当前大，那么调换位置            if (leftChildIndex < array.size && array[leftChildIndex] > array[i]) {                max = leftChildIndex            }            //如果右子节点比当前大，那么调换位置            if (rightChildIndex < array.size && array[rightChildIndex] > array[i]) {                max = rightChildIndex            }            if (max != i) {                //调换位置                swap(array, i, max)                createHeap(array, i)            }        }    }    private fun swap(array: MutableList<Int>, i: Int, max: Int) {        val temp = array[i]        array[i] = array[max]        array[max] = temp    }    fun buildHeap(array: MutableList<Int>) {        var i = array.size / 2        while (i >= 0) {            createHeap(array, i)            i--        }    }    /**     * 堆排序     */    fun sort(array: MutableList<Int>) {        var size = array.size        while (size > 0) {            buildHeap(array)            print("${array[0]} ")            //获取一次最大堆后删除后，重新生成新堆            array.removeAt(0)            size--        }    }}