Time: 76 ms (22.29%), Space: 67.6 MB (11.15%) - LeetHub

Author: hajin-kim

0973-k-closest-points-to-origin/0973-k-closest-points-to-origin.kt

@@ -0,0 +1,15 @@
+class Solution {
+    fun kClosest(points: Array<IntArray>, k: Int): Array<IntArray> {
+        val distances = points
+            .map { (x, y) ->
+                x * x + y * y
+            }
+
+        return points
+            .withIndex()
+            .sortedBy { (index, _) -> distances[index] }
+            .take(k)
+            .map { it.value }
+            .toTypedArray()
+    }
+}