build

docs/chapter_sorting/heap_sort.md

@@ -534,6 +534,44 @@ comments: true
     }
     ```
 
+=== "Kotlin"
+
+    ```kotlin title="heap_sort.kt"
+    /* 堆的长度为 n ，从节点 i 开始，从顶至底堆化 */
+    fun siftDown(nums: IntArray, n: Int, li: Int) {
+        var i = li
+        while (true) {
+            // 判断节点 i, l, r 中值最大的节点，记为 ma
+            val l = 2 * i + 1
+            val r = 2 * i + 2
+            var ma = i
+            if (l < n && nums[l] > nums[ma]) ma = l
+            if (r < n && nums[r] > nums[ma]) ma = r
+            // 若节点 i 最大或索引 l, r 越界，则无须继续堆化，跳出
+            if (ma == i) break
+            // 交换两节点
+            nums[i] = nums[ma].also { nums[ma] = nums[i] }
+            // 循环向下堆化
+            i = ma
+        }
+    }
+
+    /* 堆排序 */
+    fun heapSort(nums: IntArray) {
+        // 建堆操作：堆化除叶节点以外的其他所有节点
+        for (i in nums.size / 2 - 1 downTo 0) {
+            siftDown(nums, nums.size, i)
+        }
+        // 从堆中提取最大元素，循环 n-1 轮
+        for (i in nums.size - 1 downTo 1) {
+            // 交换根节点与最右叶节点（交换首元素与尾元素）
+            nums[0] = nums[i].also { nums[i] = nums[0] }
+            // 以根节点为起点，从顶至底进行堆化
+            siftDown(nums, i, 0)
+        }
+    }
+    ```
+
 === "Zig"
 
     ```zig title="heap_sort.zig"